5

Nx3 の配列があり、次の条件を満たす要素が必要だとします。

4 < col1 < 13, 5 > col2 >  3, 10 > col3 > 6

つまり、これを次のように適用します。

1,2,3
4,5,6
9,4,7

次に、3 つの条件すべてを満たす行のみを含む Mx3 配列を提供します。上記の例では、出力します

9,4,7

ループを実行することを考えましたが、numpy または同様のものには、はるかに高速なものが必要であると考えました (たとえば、np.where?)

4

1 に答える 1

3

列ごとにブール配列を作成します。

cond1 = (my_array[:, 0] > 4) & (my_array[:, 0] < 13)
cond2 = (my_array[:, 1] > 3) & (my_array[:, 1] < 5)
cond3 = (my_array[:, 2] > 6) & (my_array[:, 2] < 10)

次にand、3 つの条件をまとめます。

cond = cond1 & cond2 & cond3

最後に、それを使用して配列から行を選択します。

out = my_array[cond]
于 2013-06-20T19:40:29.253 に答える