1. 程式人生 > >【python】np.argmax() 返回最大值索引號

【python】np.argmax() 返回最大值索引號

import numpy as np
a = np.array([[2, 4, 6, 1], [1, 5, 2, 9]])
print(np.argmax(a))
print(np.argmax(a, axis=0))  #豎著比較,返回行號
print(np.argmax(a, axis=1))  #橫著比較,返回列號

輸出:

7
[0 1 0 1]
[2 3]