Pythonでnumpyの範囲内の要素を検索する

numpyライブラリを使用してデータを処理しているときに、特定の範囲の特定の数値を除外する必要がある場合があります。これは、numpyで利用可能ないくつかの組み込みメソッドを使用することで実現できます。

with and operator

このアプローチでは、numpy配列を取得し、それにlogical_and関数を適用します。 numpyのwhere句は、and条件を適用するためにも使用されます。結果は、必要な範囲条件を満たす要素の位置を示す配列です。

import numpy as np

A = np.array([5, 9, 11, 4, 31, 27,8])

# printing initial array
print("Given Array : ", A)

# Range 6 to 15
res = np.where(np.logical_and(A >= 6, A <= 15))

# Result
print("Array with condition : ", res)

出力

上記のコードを実行すると、次の結果が得られます-

Given Array : [ 5 9 11 4 31 27 8]
Array with condition : (array([1, 2, 6], dtype=int32),)

使用*

このアプローチでは、*演算子を使用します。配列内の値の位置ではなく、実際の値として結果が表示されます。

import numpy as np

A = np.array([5, 9, 11, 4, 31, 27,8])

# printing initial array
print("Given Array : ", A)

# Range 6 to 15
res = A [ (A >=6) * (A <= 15)]

# Result
print("Array with condition : ", res)

出力

上記のコードを実行すると、次の結果が得られます-

Given Array : [ 5 9 11 4 31 27 8]
Array with condition : [ 9 11 8]