arrays - Numpy 在所有元素之间做乘积，然后插入到一个三角形的二维数组中

Question

假设我们在下面有一个一维数组

arr = np.array([a,b,c])

我需要做的第一件事是制作所有元素的产品，即

[ab,ac,bc]

然后用这个元素构造一个二维三角形数组

[
[a,ab,ac],
[0,b,bc],
[0,0,c]
]

Answer 1

用你的一维数组创建一个对角线，并用上三角形 outer 填充它的上三角形。 :

out = np.diag(arr)
#upper triangle indices
uidx = np.triu_indices(arr.size,k=1)
#replacing upper triangle with outer
out[uidx]=np.outer(arr,arr)[uidx]