Numpy数组:每行的乘法数组

Question

我有两个数组a和b，使得b的形状是一行a。我想把b乘以a的每一行，例如，

#a
[[1,2,3]
[4,5,6]
[7,8,9]]
#b
[1,3,2]
# a times b
[[1,6,6]
[4,15,12]
[7,24,18]]

有什么办法表演吗？

Answer 1

这实际上是使用numpy时的预期输出。所以在你的情况下：

import numpy as np
a = np.array([[1,2,3], [4,5,6], [7,8,9]])
b = np.array([1,3,2])
c = a*b

c将是：

array([[ 1,  6,  6],
[ 4, 15, 12],
[ 7, 24, 18]])