当每个数组具有特定值时的 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。

Answer 2

除了a & b在 numpy 中，您还可以使用logical_and，仅供参考。np.sum(np.logical_and((x1 == 1), (y == -1)))