python古怪的乘法_python-特殊矢量和按元素乘法

我有2个数组. “ A”是任意长度的一个(假设一个起点有1000个条目),其中每个点都具有一个n维向量,其中每个条目代表一个标量. “ B”是另一个,具有n个条目,每个条目包含一个3维向量.如何进行标量乘法,以使结果为一个数组“ C”,其中每个条目是n个标量中的每个标量与n个3维向量中的每个标量的乘积？

以4-D为例：

a=[[1,2,3,4],[5,6,7,8],....]

b=[[1,0,0],[0,1,0],[0,0,1],[1,1,1]]

结果

c=[[1*[1,0,0],2*[0,1,0],3*[0,0,1],4*[1,1,1]] , [5*[1,0,0],...],...]

该实现应以numpy形式进行,而不要使用大的for循环,因为预计会有超过1000个条目.在我们的情况下,预期n为7.

解决方法:

如果您从以下内容开始：

a = np.array([[1,2,3,4],[5,6,7,8]])

b = np.array([[1,0,0],[0,1,0],[0,0,1],[1,1,1]])

然后,我们可以向a添加一个额外的轴,并沿着它重复数组就可以得到…

>>> a[:,:,None].repeat(3, axis=2)

array([[[1, 1, 1],

[2, 2, 2],

[3, 3, 3],

[4, 4, 4]],

[[5, 5, 5],

[6, 6, 6],

[7, 7, 7],

[8, 8, 8]]])

现在,就像@Jaime所说的那样,在操作时无需使用重复,因为NumPy的广播会处理它：

>>> a[:,:,None] * b

array([[[1, 0, 0],

[0, 2, 0],

[0, 0, 3],

[4, 4, 4]],

[[5, 0, 0],

[0, 6, 0],

[0, 0, 7],

[8, 8, 8]]])

标签：arrays,python,numpy