版权声明:本文为博主原创文章,转载请附上博文链接! https://blog.csdn.net/coco_link/article/details/88313474
获取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]))