python - Numpy 列减法；从 (k,) 数组中的值减去 (k,n) 数组

Question

说我有

x=np.random.random((3,10))
y=np.array([1,10,100])

我想从 y 中减去 x 每一列中的值。我可以这样做:

np.array([y]*10).T-x

但是，这涉及创建一个大小为 y 10 倍的新数组(在本例中)。我也可以想象使用 for 循环。任何人都可以建议更好的方法吗？

Answer 1

只需垂直堆叠 y 并减去 x。例如:

y[:, None] - x    # insert a new axis into y (creating a new view of y)

或

np.vstack(y) - x  # stack y vertically (creating a copy of y)

这是可行的，因为 y 现在具有 (3, 1) 的形状，因此可以使用具有 形状的 x 进行广播(3, 10)。

如果 x 的每个尾部维度(即从形状元组的末尾开始)相等，则

x 和 y 可以一起广播到 y 的尾部维度，或者比较的维度之一为 1。在本例中，y 的尾部维度更改为 1，因此两个数组是兼容的。