web-dev-qa-db-ja.com

numpy.whereを論理演算子で使用する方法

私は、配列の中でaより大きくbより小さいすべての要素のインデックスを見つけようとしています。それはおそらく私の構文に関する問題ですが、これは機能しません：

numpy.where((my_array > a) and (my_array < b))

どうすれば修正できますか？またはそれを行うためのより良い方法はありますか？

ありがとう！

pythonnumpywherelogical-operators
40
rferdinand

次の2つの方法があります。

In [1]: my_array = arange(10)

In [2]: where((my_array > 3) & (my_array < 7))
Out[2]: (array([4, 5, 6]),)

In [3]: where(logical_and(my_array > 3, my_array < 7))
Out[3]: (array([4, 5, 6]),)

最初の（andを&に置き換える）、括弧を適切に追加するよう注意してください：&は比較演算子よりも優先順位が高くなります。 *を使用することもできますが、お勧めしません。それはハックで、読み取り可能なコードを作成しません。

In [4]: where((my_array > 3) * (my_array < 7))
Out[4]: (array([4, 5, 6]),)
60
Mark Dickinson