numpy里面的argmax函数
函数原型:
def argmax(a, axis=None, out=None)
a----输入array
axis----为0代表列方向,为1代表行方向
out----结果写到这个array里面
例子:
>>>import numpy as np a = np.array([[2,4,6,1],[1,5,2,9]]) >>>print a [[2 4 6 1] [1 5 2 9]] >>>np.argmax(a) ;;返回array中数值最大数的下标,默认将输入array视作一维,出现相同的最大,返回第一次出现的 7 >>>np.argmax(a, axis=0) ;;返回列方向上数值最大值下标,2>1, 4<5, 6>2, 1<9所以结果为0, 1, 0, 1 array([0, 1, 0, 1]) >>>np.argmax(a, axis=1) ;;返回行方向上数值最大值下标,同理axis = 0情况 array([2, 3])
————————————————
原文链接:https://blog.csdn.net/banana1006034246/article/details/73331420