numpy.nonzero()与组合逻辑

时间:2015-03-07 15:37:12

标签: python numpy

我想将numpy.nonzero()与组合逻辑一起使用,但我的尝试最终会像这样:

>>> x
array([[  3,   5,   4,   2,   2],
       [  2,   5,   3, 100,   4],
       [  3,   5,   4, 100,   3]])
>>> np.nonzero(x > 3 and x < 100)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

但在这种情况下,不清楚如何应用any()或all()。 我是否必须分成两个步骤,然后使用setdiff()? 那会很难过。

1 个答案:

答案 0 :(得分:3)

您需要使用运营商&amp;得到你期望的结果:

np.nonzero((x > 3) & (x < 100))

&安培;是元素方式&#34;和&#34;