向量的所有元素之间的逐元素乘法

Question

在numpy数组的所有元素之间进行元素明智的乘法的有效方法是什么。

如果A是100个元素的向量，则A2是100 ^ 2个元素的向量。

示例

输入：

a = [5,2,3,4]

输出

a2=[5*5,5*2,5*3,5*4,2*5,2*2,2*3,2*4,3*5,3*2,3*3,3*4,4*5,4*2,4*3,4*4]

谢谢

Answer 1

>>> import numpy as np
>>> np.outer(a,a).reshape(-1)

array([25, 10, 15, 20, 10,  4,  6,  8, 15,  6,  9, 12, 20,  8, 12, 16])

使用numpy.outer-计算两个向量的外积 然后numpy.reshape-使数组变成预期的形状。