numpy 中 argmax 返回的是最大值的索引
对于一维数组来说
import numpy as np
a = np.array([4, 9, 17, 3, 5])
print(np.argmax(a))
2
对于二维数组来说
(1)当你要找每一列的最大值的索引时
import numpy as np
a = np.array([[4, 9, 17, 5],
[8, 1, 13, 7],
[5, 10, 9, 6]])
print(np.argmax(a,axis=0))
[1 2 0 1]
(2)当你要找每一行的最大值的索引时
import numpy as np
a = np.array([[4, 9, 17, 5],
[8, 1, 13, 7],
[5, 10, 9, 6]])
print(np.argmax(a,axis=1))
[2 2 1]