获取numpy数组最大元素的索引

获取numpy数组最大元素

a = np.arange(9).reshape((3,3))
Out[4]: 
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

np.max(a)  # 全局最大
Out[5]: 8

np.max(a,axis=0)   # 每列最大
Out[6]: array([6, 7, 8])

np.max(a,axis=1)  # 每行最大
Out[7]: array([2, 5, 8])

获取numpy数组最大元素的索引

使用where函数得到最大值的索引，返回值中，前面的array对应行数，后者对应列数

np.where(a==np.max(a))
Out[8]: (array([2]), array([2]))  #表示最大值在第二行第二列

np.where(a==np.max(a,axis=0))
Out[9]: (array([2, 2, 2]), array([0, 1, 2]))  # 表示最大值分别在第二行第零列，第二行第一列，第二行第二列

如果array中有相同的最大值，where会将其位置全部给出

a[1,0]=8
a
Out[11]: 
array([[0, 1, 2],
       [8, 4, 5],
       [6, 7, 8]])
       
np.where(a==np.max(a))
Out[12]: (array([1, 2]), array([0, 2]))