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

Question

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

Answer 1

如果使用列表理解，则可以比直接循环更高效：

import numpy as np

a = np.array([10, 20, 30, 40])
b = np.array([1, 2])


c = np.array([a - b[j] for j in range(len(b))])
print(c)

输出：

^{pr2}$