python - 如何使用Python在满足两个条件的数组中查找值

Question

我有一个数组

a=[1,2,3,4,5,6,7,8,9]

我想找到满足两个条件的元素 s 的索引，即

a>3 and a<8
ans=[3,4,5,6]
a[ans]=[4,5,6,7]

我可以使用 numpy.nonzero(a>3)或 numpy.nonzero(a<8)但不是 numpy.nonzero(a>3 and a<8)这给出了错误:

ValueError: The truth value of an array with more than one element is
ambiguous. Use a.any() or a.all()

当我尝试使用 any或 all我犯了同样的错误。是否可以结合两个条件测试来获得答案？

Answer 1

numpy.nonzero((a > 3) & (a < 8))

& 执行元素级 bool 值与。