在 numpy 数组中相乘

我试图将二维数组中的每个项乘以一维数组中的相应项。如果我想将每一列乘以一维数组，这非常容易，如下所示numpy.乘法功能。但我想做相反的事情，将行中的每一项相乘。 换句话说，我想乘以：

[1,2,3]   [0]
[4,5,6] * [1]
[7,8,9]   [2]

and get

[0,0,0]
[4,5,6]
[14,16,18]

但我得到

[0,2,6]
[0,5,12]
[0,8,18]

有谁知道是否有一种优雅的方法可以用 numpy 来做到这一点？ 多谢， 亚历克斯

正常乘法就像你展示的那样：

>>> import numpy as np
>>> m = np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> c = np.array([0,1,2])
>>> m * c
array([[ 0,  2,  6],
       [ 0,  5, 12],
       [ 0,  8, 18]])

如果添加一个轴，它将按照您想要的方式相乘：

>>> m * c[:, np.newaxis]
array([[ 0,  0,  0],
       [ 4,  5,  6],
       [14, 16, 18]])

你也可以转置两次：

>>> (m.T * c).T
array([[ 0,  0,  0],
       [ 4,  5,  6],
       [14, 16, 18]])