有两种方法:
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
用&
),小心添加适当括号:&
具有比比较运算符更高的优先级。您也可以使用*
,但我不推荐这样做:它很hacky,不适合可读代码。
In [4]: where((my_array > 3) * (my_array < 7))
Out[4]: (array([4, 5, 6]),)