条件式を指定するNumPyのwhereの使い方

arr[a < 10]のように、インデックス部分に条件を指定することで、目的の要素を取得することができました。np.whereを使うことで、値ではなくインデックスを取得することができます。 条件の指定 基本的な使い方は、第一引数に条件のみを指定する方法です。以下のように第一引数に条件式を指定することで、条件を満たす要素のインデックスを取得することができます。 In [1]: import numpy as np In [2]: a = np.arange(20, 0, -2) # まずは1次元配列を生成 In [3]: a Out[3]: array([20, 18, 16, 14, 12, 10, 8, 6, 4, 2]) In [4]: np.where(a < 10) # 10未満のindexを取得 Out[4]: (array([6, 7, 8, 9]),) In [5]: a[n