Numpy tips

提供: Eospedia
2020年10月27日 (火) 04:55時点におけるKttn (トーク | 投稿記録)による版

(差分) ←前の版 | 最新版 (差分) | 次の版→ (差分)
移動: 案内検索

ndarray

特定の値を持つ要素のインデックスを得る

例えばhogeというndarrayの中で値 0 をもつ要素のインデックスを得るには、

np.where(hoge == 0)[0]

とすればよい。np.whereはタプルを返してきて、その最初の要素が当該インデックスのndarrayになっている。

これは以下の式と等価。(hogeがndarrayであれば。)

(hoge == 0).nonzero()