如何将矩阵中的每一行乘以它在 NumPy 中的标量?
How to multiply each row in matrix by its scalar in NumPy?
例如我有一个矩阵和一个标量向量
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
您可以使用广播:A * B[:, None]
:
array([[ 1, 2, 3],
[ 8, 10, 12],
[21, 24, 27]])
例如我有一个矩阵和一个标量向量
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
您可以使用广播:A * B[:, None]
:
array([[ 1, 2, 3],
[ 8, 10, 12],
[21, 24, 27]])