【问题标题】:Element-wise subtraction of two numpy arrays两个numpy数组的元素减法
【发布时间】:2015-11-19 00:26:53
【问题描述】:

我有两个 numpy 数组,分别是尺寸为 (327600,5) 和 (3,5) 的“a”和“b”。我想对两个数组进行逐元素减法。我将如何在没有循环的情况下执行此操作?

当我执行 a[0] - b 时,我得到一个 (3,5) 数组。我想得到一个 (327600,3,5) 数组作为减法后的结果。

【问题讨论】:

    标签: python arrays numpy vectorization


    【解决方案1】:
    a[:, np.newaxis] - b
    

    您需要查看 a 的形状,并在形状中添加一个额外的长度为 1 的轴,以便轴对齐以进行广播。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2015-05-13
      相关资源
      最近更新 更多