Numpy在这里函数多个条件

我有一个名为dist的距离数组。我想选择一个范围内的dist。

 dists[(np.where(dists >= r)) and (np.where(dists <= r + dr))]

但是，这只选择条件

 (np.where(dists <= r + dr))

如果我使用临时变量按顺序执行命令，它可以很好地工作。为什么上面的代码不能工作，我如何让它工作?

当前回答

公认的答案很好地解释了这个问题。但是，应用多个条件的更Numpythonic方法是使用numpy逻辑函数。在这种情况下，您可以使用np.logical_and:

np.where(np.logical_and(np.greater_equal(dists,r),np.greater_equal(dists,r + dr)))

其他回答

公认的答案很好地解释了这个问题。但是，应用多个条件的更Numpythonic方法是使用numpy逻辑函数。在这种情况下，您可以使用np.logical_and:

np.where(np.logical_and(np.greater_equal(dists,r),np.greater_equal(dists,r + dr)))

Try:

import numpy as np
dist = np.array([1,2,3,4,5])
r = 2
dr = 3
np.where(np.logical_and(dist> r, dist<=r+dr))

输出:(数组([2,3,4])，)

您可以查看逻辑函数了解更多详细信息。

Try:

np.intersect1d(np.where(dists >= r)[0],np.where(dists <= r + dr)[0])

这应该可以工作:

dists[((dists >= r) & (dists <= r+dr))]

要让np.where()在多个条件下工作，只需执行以下操作:

np.where((condition 1) & (condition 2)) # for and
np.where((condition 1) | (condition 2)) # for or

我知道这重复了一些其他的答案，但我把这个简单的答案放在这里给那些仍然疑惑的人，“为什么我收到关于一个包含多个元素的数组的真值是模棱两可的烦人的错误消息”，他们被非常冗长和复杂的回答所迷惑，这些回答解决了原始帖子的一些专门化性质。

现在，至于为什么numpy会在使用and而不是&时中断，我不会在这里回答这个问题。它就是这样:)在这里看到其他答案的解释。在我看来，这似乎是他们应该解决的问题，而不是为了一致性而强迫它。或者至少他们应该给出更好的错误消息。：）

推荐文章