Numpyのwhereで配列インデックスを取得(python)

ndarray 配列と where についてです。

条件にマッチするインデックスを取得

値を取得

ndarray 配列から任意の条件で値を取り出すには [] の中に条件を書いて

インデックスを取得

一方、値ではなくインデックスがほしい場合は where を使います。

ちなみに (data < 300).nonzero() でも同じくインデックスを取得できますが

where のほうがパッと見わかりやすい気がします。

3項演算子ふうに

where で3項演算子のようなこともできます。
引数は3つ

  1. 条件
  2. 真の場合の値
  3. 偽の場合の値

コメントはお気軽に