代码之家  ›  专栏  ›  技术社区  ›  Enryu

当每个数组都有一个特定值时的Numpy Sum

  •  2
  • Enryu  · 技术社区  · 6 年前
    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])
    

    我知道在这两个数组中,你可以求出这行代码中精确指标与numpy相等的次数。

    np.sum(x1 == y)
    

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

    不幸的是,这行代码没有运行,但如果它工作的结果应该是3。

    2 回复  |  直到 6 年前
        1
  •  7
  •   michcio1234    6 年前

    你只要用一个 & 再加上一些括号:

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

    结果得到3。

        2
  •  1
  •   atline    6 年前

    除此之外 a & b 在numpy中,您还可以使用 logical_and ,仅供参考。

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