やること
X = np.array([[1, 2, 3],[4,100, 6]])
から100のindexである (1, 1)
を取得したい。
解決法
np.argmaxを使う
np.argmax(X)
>> 5
と 行列を一次元にreshapeした時のindexが返ってくる。
np.whereを使う
zip(np.where(X==X.max())[0], np.where(X==X.max())[1]) >> [(1, 1)]
とすれば最大値がある indexが取得できる。