112

たとえば、数字の配列がたくさんあります。 

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

特定の範囲内の要素のすべてのインデックスを検索したいと思います。たとえば、範囲が（6、10）の場合、答えは（3、4、5）になります。これを行うための組み込み関数はありますか？

4

12 に答える 12

185

を使用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]),)
于 2012-12-13T22:01:04.937 に答える
73

@deinonychusaurの返信と同じですが、さらにコンパクトです。

In [7]: np.where((a >= 6) & (a <=10))
Out[7]: (array([3, 4, 5]),)
于 2012-12-14T02:55:05.097 に答える
17

aあなたが与えた例ではソートされているので、これを追加すると思いました：

import numpy as np
a = [1, 3, 5, 6, 9, 10, 14, 15, 56] 
start = np.searchsorted(a, 6, 'left')
end = np.searchsorted(a, 10, 'right')
rng = np.arange(start, end)
rng
# array([3, 4, 5])
于 2012-12-13T23:59:29.840 に答える
0 
s=[52, 33, 70, 39, 57, 59, 7, 2, 46, 69, 11, 74, 58, 60, 63, 43, 75, 92, 65, 19, 1, 79, 22, 38, 26, 3, 66, 88, 9, 15, 28, 44, 67, 87, 21, 49, 85, 32, 89, 77, 47, 93, 35, 12, 73, 76, 50, 45, 5, 29, 97, 94, 95, 56, 48, 71, 54, 55, 51, 23, 84, 80, 62, 30, 13, 34]

dic={}

for i in range(0,len(s),10):
    dic[i,i+10]=list(filter(lambda x:((x>=i)&(x<i+10)),s))
print(dic)

for keys,values in dic.items():
    print(keys)
    print(values)

出力： 

(0, 10)
[7, 2, 1, 3, 9, 5]
(20, 30)
[22, 26, 28, 21, 29, 23]
(30, 40)
[33, 39, 38, 32, 35, 30, 34]
(10, 20)
[11, 19, 15, 12, 13]
(40, 50)
[46, 43, 44, 49, 47, 45, 48]
(60, 70)
[69, 60, 63, 65, 66, 67, 62]
(50, 60)
[52, 57, 59, 58, 50, 56, 54, 55, 51]
于 2016-07-08T06:27:13.147 に答える