【Pythonメモ】numpyメソッド argmax

argmax

配列の中の最大値のインデックスを返すメソッドです。

>>> import numpy as np 
>>> test = np.array([1,2,5,40,20,30]) #配列は6個
>>> np.argmax(test)
3 #インデックスは0から数える

>>> test2 = np.array([[1,2,3][10,20,30]]) #二次配列
>>> np.argmax(test2) #全要素の中から最大値を返す
5  

※軸を指定する時は第二引数にaxis=2といった形で指定

>>> test3 = [[1,2,5],[40,20,30],[10,20,30]]
>>> np.argmax(test3,axis=1)　#軸を1にしあｔ
array([2, 0, 2])

>>> test4 =[[[1,2,5],[40,20,30],[10,20,30]],[[1,2,5],[40,20,30],[10,20,30]]]
>>> np.argmax(test4,axis=2) #軸を2にした
array([[2, 0, 2],
       [2, 0, 2]])