python - 将 numpy 二维数组与一维数组相乘

Question

我有一个 numpy 二维数组:

[[1,1,1],
[1,1,1],
[1,1,1],
[1,1,1]]

当二维数组的行长度小于一维数组的长度时，如何才能将从上到下的索引与一维数组中的相应值相乘？例如，将上面的内容与此相乘:

[10, 20, 30, 40]

要得到这个:

 [[10, 10, 10],
 [20, 20, 20],
 [30, 30, 30]
 [40, 40, 40]]

可能是重复的，但我找不到我正在寻找的确切内容。提前致谢。

Answer 1

numpy 中的

* 进行逐元素乘法，例如将一维数组乘以另一个一维数组:

In [52]: np.array([3,4,5]) * np.array([1,2,3])
Out[52]: array([ 3,  8, 15])

当你将二维数组乘以一维数组时，二维数组的每一行都会发生同样的事情:

In [53]: np.array([[3,4,5],[4,5,6]]) * np.array([1,2,3])
Out[53]:
array([[ 3,  8, 15],
       [ 4, 10, 18]])

对于您的具体示例:

In [66]: ones = np.ones(12, dtype=np.int).reshape(4,3)

In [67]: a = np.array([10, 20, 30, 40])

In [68]: (ones.T * a).T
Out[68]:
array([[10, 10, 10],
       [20, 20, 20],
       [30, 30, 30],
       [40, 40, 40]])