python - Numpy Sum 当每个数组都有特定值时

Question

x1 = np.array([0,  1,  0,  1,  0, 1, 0, 1, 1,  1])
y = np.array([-1, -1, -1, -1, -1, 1, 1, 1, 1, -1])

我知道使用这 2 个数组，您可以在这行代码中计算精确索引与 numpy 相等的次数。

np.sum(x1 == y)

但是有没有一种方法可以在每次相同的索引等于每个数组上的特定值时进行总结，例如

np.sum(x1 == 1 && y == -1)

不幸的是，这行代码没有运行，但如果运行成功，结果应该是 3。

Answer 1

你只需要使用一个 & 并添加一些括号:

np.sum((x1 == 1) & (y == -1))

结果是 3。