読者です 読者をやめる 読者になる 読者になる

ゆるふわめも

in Kyoto or Tokyo

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