11
我想要生成一個數組,每一行的最大值最大。確定Python的NumPy中最大值的索引
a = np.array([ [1,2,3], [6,5,4], [0,1,0] ])
maxIndexArray = getMaxIndexOnEachRow(a)
print maxIndexArray
[[2], [0], [1]]
有一個np.argmax功能,但它似乎並沒有做我想做的......
我想要生成一個數組,每一行的最大值最大。確定Python的NumPy中最大值的索引
a = np.array([ [1,2,3], [6,5,4], [0,1,0] ])
maxIndexArray = getMaxIndexOnEachRow(a)
print maxIndexArray
[[2], [0], [1]]
有一個np.argmax功能,但它似乎並沒有做我想做的......
的argmax()
功能不做你想要什麼:
print a.argmax(axis=1)
array([2, 0, 1])
蟒真棒 – JohnTortugo 2013-11-10 19:46:14
這也作爲一個頂級函數存在,因此下面也有效:'maxvals = np.argmax(a,axis = 1)' – Robin 2014-04-03 14:24:00