numpy / scipy：用一维向量的元素填充数组的上三角吗？

Question

假设我有一个扁平化的1D向量，它与2D数组的上三角元素完全对应。

需要将1D向量读入上三角。

我可能会在python中执行以下操作：

triu_flat = ...
row,col = np.triu_indices(50)

D = np.zeros((50,50))

i=0 
for r in row: 
    for c in col:
        D[r,c] = triu_flat[i]
        i++

但是，必须有一种通过numpy / scipy操作执行此操作的方法。

Answer 1

您可以简单地使用triu_indices()返回的索引，而无需for循环：

import numpy

data = numpy.arange(6)
out = numpy.zeros((3, 3))
inds = numpy.triu_indices(len(out))
out[inds] = data