我想知道是否有一种有效的方法来执行以下操作。我有两个(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'。我当前的解决方案很丑陋,希望得到任何建议。
最佳答案
您可以使用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
的子类。
关于python - 基于另一个数组对一个数组中的实例进行计数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49675331/