Numpy:如何逐个乘以两个向量,shape(n,1)和(n,)？

Question

如果它们都具有相同的形状,例如两个(n,1)或两个(n,),则两个矢量的元素相乘是没有问题的.但是,如果一个向量具有形状(n,1)而另一个向量具有(n,),则*-operator返回一些有趣的东西.

a = np.ones((3,1))
b = np.ones((3,))
print a * b

得到的nxn矩阵包含A_ {i,j} = a_i*b_j.

我怎样才能实现elementwise乘法的a和b呢？

Answer 1

以使形状匹配的方式切割向量:

a[:, 0] * b

要么

a * b[:, None]