我在 numpy 中有两个数组 A 和 B。 A 保存笛卡尔坐标,每一行是 3D 空间中的一个点,形状为 (r, 3)。 B 的形状为 (r, n) 并包含整数。
我想做的是将 B 的每个元素与 A 中的每一行相乘,以便生成的数组具有形状 (r, n, 3)。例如:
# r = 3
A = np.array([1,1,1, 2,2,2, 3,3,3]).reshape(3,3)
# n = 2
B = np.array([10, 20, 30, 40, 50, 60]).reshape(3,2)
# Result with shape (3, 2, 3):
# [[[10,10,10], [20,20,20]],
# [[60,60,60], [80,80,80]]
# [[150,150,150], [180,180,180]]]
我很确定这可以用 np.einsum
来完成,但我已经尝试了很长时间,但无法正常工作。
最佳答案
使用broadcasting
-
A[:,None,:]*B[:,:,None]
自 np.einsum
还支持广播,你也可以使用它(感谢@ajcr 推荐这个简洁的版本)-
np.einsum('ij,ik->ikj',A,B)
sample 运行-
In [22]: A
Out[22]:
array([[1, 1, 1],
[2, 2, 2],
[3, 3, 3]])
In [23]: B
Out[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]]])
关于python - 将一个数组的每一行与numpy中另一个数组的每个元素相乘,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32561332/