python - 如何从另一个数组中减去一个数组的每个元素？

Question

我有两个不同大小的 Numpy 数组 A (n x 1) 和 B (m x 1)。我想从 A 的所有元素中减去 B 的每个元素。因此，结果矩阵 C (m x n) 的元素应计算为 c(i,j) = A(i)-B(j)。有没有使用 Numpy 的直接无循环计算？

Answer 1

广播:

A = np.array([1,2,3,4,5])
B = np.array([5,4,2,7])
A - B[:, np.newaxis]

输出:

array([[-4, -3, -2, -1,  0],
       [-3, -2, -1,  0,  1],
       [-1,  0,  1,  2,  3],
       [-6, -5, -4, -3, -2]])