>> import numpy as np
>> a = np.array([[5, 50, 10],[20,-10,100]])
>> np.argmax(a) # 最大値のインデックス
5
>> np.argmin(a) # 最小値のインデックス
4
>> np.argmax(a,axis=0) # 列同士での最大値
array([1, 0, 1], dtype=int64)
>> np.argmax(a,axis=1) # 行同士での最大値
array([1, 2], dtype=int64)