如何使用Python在满足两个条件的数组中查找值
问题描述
我有一个数组
a=[1,2,3,4,5,6,7,8,9]
我想找到满足两个条件的元素 s 的索引，即
and I want to find the indices of the element s that meet two conditions i.e.
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) 给出错误:
I can use numpy.nonzero(a>3) or numpy.nonzero(a<8)
but not
numpy.nonzero(a>3 and a<8) which gives the error:
ValueError: The truth value of an array with more than one element is
ambiguous. Use a.any() or a.all()
当我尝试使用 any 或 all 时，我得到了同样的错误.是否可以结合两个条件测试来获得答案?
When I try to use any or all I get the same error.
Is it possible to combine two conditional tests to get the ans?
解决方案
numpy.nonzero((a > 3) & (a < 8))
&做一个元素方式的布尔值和.
& does an element-wise boolean and.
相关文章