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

How find values in an array that meet two conditions using Python(如何使用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.

这篇关于如何使用Python在满足两个条件的数组中查找值的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！