如何检查向量的元素是否位于 Numpy 中的两个向量之间？

Question

假设A和B被赋予向量，目的是检查Y哪些元素位于 A 和 B 之间。例如：

A=np.array([1,2,3,4])
B=np.array([10,20,30,40])
Y=np.array([8,15,0,50])

预期输出应如下所示： [1,1,0,0]其中元素不是 bool 类型，因此我可以使用np.sum()找出真实值的数量

Answer 1

您可以进行比较并进行元素and

(A < Y) & (Y < B)

无论它们是布尔值， np.sum()都可以工作。 如有疑问，只需使用 int 强制转换

X.astype(int)

Answer 2

就这么简单

np.logical_and(A <= Y, Y <= B).astype(int)

但是你也可以对一个逻辑向量求和，numpy 将处理引擎盖下的转换。

In [1]: np.sum(np.logical_and(A <= Y, Y <= B).astype(int)) == np.sum(np.logical_and(A <= Y, Y <= B))
Out[1]: True