めも

メモ.

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

プライバシーポリシー

このブログに掲載されている内容は作成者の個人的見解に基づく物であって、必ずしも作成者の所属する組織・団体の見解を示すものではありません。また、記載が不正確であったことにより生じたいかなる損害に関しても、責任を負いかねますのでご了承ください。また、本サイトは、Amazon.co.jpを宣伝しリンクすることによってサイトが紹介料を獲得できる手段を提供することを目的に設定されたアフィリエイトプログラムである、Amazonアソシエイト・プログラムの参加者です。また、本サイトでは、第三者配信の広告サービス(Googleアドセンス、A8.net)を利用しており、ユーザーの興味に応じた商品やサービスの広告を表示するため、クッキー(Cookie)を使用しております。 クッキーを使用することで当サイトはお客様のコンピュータを識別できるようになりますが、お客様個人を特定できるものではありません。本サイトの管理者への問い合わせ、当ブログのプライバシーポリシーの詳細についてはこちらをご覧ください。