python - 基于另一个数组对一个数组中的实例进行计数

Question

我想知道是否有一种有效的方法来执行以下操作。我有两个(numpy)数组，我想根据另一个数组的条件来计算一个数组中出现的值的实例数。例如:

a = np.array([1,-1,1,1,-1,-1])
b = np.array([.75,.35,.7,.8,.2,.6])

我想将 c 计算为当 b > .5 时 a 中出现的 1 的数量，因此在本例中` c = 3'。我当前的解决方案很丑陋，希望得到任何建议。

Answer 1

您可以使用numpy.sum来实现:

a = np.array([1,-1,1,1,-1,-1])
b = np.array([.75,.35,.7,.8,.2,.6])

np.sum((a == 1) & (b > .5))  # 3

这是有效的，因为 bool 是 int 的子类。