argmaxで、最大値のインデックスをとる
>>> import numpy as np >>> a = np.array([1,2,3,4,5]) >>> np.argmax(a) 4
最大値な値が複数あったら、最初のインデックスをとる
>>> b = np.array([3,3,1,2,0]) >>> np.argmax(b) 0
argmaxで、最大値のインデックスをとる
>>> import numpy as np >>> a = np.array([1,2,3,4,5]) >>> np.argmax(a) 4
最大値な値が複数あったら、最初のインデックスをとる
>>> b = np.array([3,3,1,2,0]) >>> np.argmax(b) 0