【问题标题】:Numpy Sum when each array has a specific value当每个数组具有特定值时的 Numpy Sum
【发布时间】:2018-10-10 07:36:35
【问题描述】:
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。

【问题讨论】:

    标签: python numpy


    【解决方案1】:

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

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

    结果为 3。

    【讨论】:

      【解决方案2】:

      除了 numpy 中的a & b,您还可以使用logical_and,仅供参考。

      np.sum(np.logical_and((x1 == 1), (y == -1)))
      

      【讨论】:

        猜你喜欢
        • 1970-01-01
        • 1970-01-01
        • 2019-03-11
        • 2013-07-12
        • 1970-01-01
        • 2019-01-16
        • 1970-01-01
        • 1970-01-01
        • 2022-11-03
        相关资源
        最近更新 更多