python - 如何将矩阵中的每一行乘以其在 NumPy 中的标量？

Question

例如，我有一个矩阵和一个标量向量

A = np.array([
    [1,2,3],
    [4,5,6],
    [7,8,9]
])

B = np.array([1, 2, 3])

我想得到的输出是这样的，基本上我想将每一行乘以具有相同索引的标量

 1   2   3
 8  10  12
21  24  27

标签： pythonnumpylinear-algebra

Answer 1

您可以使用广播: A * B[:, None]:

array([[ 1,  2,  3],
       [ 8, 10, 12],
       [21, 24, 27]])