將1D numpy數組轉換爲矩陣的頂部三角形

Question

是否可以採用包含171個值的1D numpy數組並將這些值放入18X18矩陣的頂部三角形中？

我覺得我應該能夠使用np.triu_indices來做到這一點，但我不能弄明白！

謝謝！

Answer 1

見What is the most efficient way to get this kind of matrix from a 1D numpy array?和Copy flat list of upper triangle entries to full matrix?

大致的做法是

result = np.zeros(...) 
ind = np.triu_indices(...) 
result[ind] = values 

細節取決於目標數組的大小，和你的價值觀目標的佈局。

In [680]: ind = np.triu_indices(4) 
In [681]: values = np.arange(16).reshape(4,4)[ind] 
In [682]: result = np.zeros((4,4),int) 
In [683]: result[ind]=values 
In [684]: values 
Out[684]: array([ 0, 1, 2, 3, 5, 6, 7, 10, 11, 15]) 
In [685]: result 
Out[685]: 
array([[ 0, 1, 2, 3], 
     [ 0, 5, 6, 7], 
     [ 0, 0, 10, 11], 
     [ 0, 0, 0, 15]])