形状为（N，M）和（N，）的数组之间的广播操作

Question

这是一个相当简单的问题，但我似乎找不到答案。 考虑两个简单的数组：

import numpy as np
a = np.random.uniform(0., 1., (2, 1000))
b = np.random.uniform(0., 1., (2,))

我想要执行操作a - b以便最终数组为([[a[0] - b[0], a[1] - b[1]]) ，我得到

ValueError: operands could not be broadcast together with shapes (2,1000) (2,) 

如何执行此（或其他）操作？

Answer 1

根据一般广播规则 ：

在两个数组上进行操作时，NumPy逐元素比较其形状。 它从尾随尺寸开始，一直向前发展。 两种尺寸兼容

1. 它们相等，或者
2. 其中之一是1

出现错误是因为无法广播a （1000）和b （2）的最后维度； 你可以转换b到形状的2D阵列(2, 1)使得1 - >（可广播到） 1000 ， 2 - >（可广播到） 2 ：

a - b[:,None]                            
#array([[ 0.06475683, -0.43773571, -0.62561564, ...,  0.05205518,
#        -0.1209487 ,  0.16334639],
#       [ 0.58443617,  0.28764136,  0.75789299, ...,  0.18159133,
#         0.28548633, -0.12037869]])

要么

a - b.reshape(2,1)