python - 两个 numpy 数组的逐元素减法

Question

我有两个 numpy 数组 'a' 和 'b' 分别具有 (327600,5) 和 (3,5) 的维度。我想对两个数组进行逐元素减法。如果没有循环，我将如何去做呢？

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

Answer 1

a[:, np.newaxis] - b

您需要查看 a 在形状中有一个额外的 length-1 轴，以便这些轴对齐以进行广播。