如何将数组的行乘以存储在向量的相应行中的值？

Question

我有以下设置：

a = np.asarray([[1, 2, 3], [4, 5, 6]])
s = np.asarray([1, 2])

我想将a[0] * s[0]和a[1] * s[1]相乘得到

[[1, 2, 3],
 [8, 10, 12]]

我怎样才能轻松完成这项工作？如果可能的话，我想避免重塑。

Answer 1

a * s[:, np.newaxis]

广播通过与你想要的相反方向的形状，所以我们在那里粘贴另一个轴。 （这非常类似于重塑。）