python - 使用广播根据向量中的元素乘以矩阵行？

Question

假设我有一个矩阵

x=array([[ 0.,  0.,  0.],
       [ 0.,  0.,  1.],
       [ 0.,  1.,  0.],
       [ 0.,  1.,  1.],
       [ 1.,  0.,  0.],
       [ 1.,  0.,  1.],
       [ 1.,  1.,  0.],
       [ 1.,  1.,  1.]])

我想得到

array([[ 0.,  0.,  0.],
       [ 0.,  0.,  2.],
       [ 0.,  3.,  0.],
       [ 0.,  4.,  4.],
       [ 5.,  0.,  0.],
       [ 6.,  0.,  6.],
       [ 7.,  7.,  0.],
       [ 8.,  8.,  8.]])

如何写x和range(1,9)之间的一行表达式？列的相同操作的代码是什么？

Answer 1

x * np.arange(1, 9).reshape(-1, 1)

或

x * arange(1, 9)[:, np.newaxis]

这两种形式都从 arange(1, 9) 中生成一个列向量，它沿 x 的 y 轴很好地传播。

“列的相同操作”只是上面的转置，即跳过reshape操作:

x * arange(1, 4)