☰

Numpy: 範囲内の要素のインデックスを見つける 質問する

tag-icon • tag-icon python numpy
Numpy: 範囲内の要素のインデックスを見つける 質問する

たとえば、NumPyの数値配列があります。

a = np.array([1, 3, 5, 6, 9, 10, 14, 15, 56])

特定の範囲にある要素のインデックスをすべて見つけたいと思います。たとえば、範囲が (6, 10) の場合、答えは (3, 4, 5) になります。これを行う組み込み関数はありますか?

ベストアンサー1

np.whereインデックスを取得し、np.logical_and2 つの条件を設定するために使用できます。

import numpy as np
a = np.array([1, 3, 5, 6, 9, 10, 14, 15, 56])

np.where(np.logical_and(a>=6, a<=10))
# returns (array([3, 4, 5]),)

おすすめ記事