将一个数组的每一行与另一个数组的每个元素乘以numpy

将一个数组的每一行与另一个数组的每个元素乘以numpy

使用

broadcasting

-

A[:,None,:]*B[:,:,None]

由于

np.einsum

还支持广播，因此您也可以使用它（感谢@ajcr提出了这个简洁的版本）-

np.einsum('ij,ik->ikj',A,B)

样品运行-

In [22]: AOut[22]: array([[1, 1, 1],       [2, 2, 2],       [3, 3, 3]])In [23]: BOut[23]: array([[10, 20],       [30, 40],       [50, 60]])In [24]: A[:,None,:]*B[:,:,None]Out[24]: array([[[ 10,  10,  10],        [ 20,  20,  20]],       [[ 60,  60,  60],        [ 80,  80,  80]],       [[150, 150, 150],        [180, 180, 180]]])In [25]: np.einsum('ijk,ij->ijk',A[:,None,:],B)Out[25]: array([[[ 10,  10,  10],        [ 20,  20,  20]],       [[ 60,  60,  60],        [ 80,  80,  80]],       [[150, 150, 150],        [180, 180, 180]]])