python - 避免使用 numpy 矩阵进行循环

Question

我想知道这一定是一种在没有 for 循环的情况下执行以下操作的方法:

import numpy
from itertools import product as itprod

a = np.arange(120.).reshape(3,2,5,2,2)
fact = np.linspace(1,1.4,15).reshape((3,5))

for i,j in itprod(range(3),range(5)):
    a[i,:,j]*= fact[i,j]

有什么建议吗？？

Answer 1

要利用广播，您必须在正确的位置为 fact 插入新轴:

a *= fact[:, np.newaxis, :, np.newaxis, np.newaxis]