ゆるふわめも

東京か京都にいます。

pythonでnp.array(二次元)の最大値があるindexを取得する。

やること

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が取得できる。