Numpy:将数组转换为三角矩阵

Question

我正在寻找一种将线性数组转换为三角矩阵的内置方法。由于我没有找到一个，我请求帮助实施一个。

想象一个数组：

In [203]: dm
Out[203]: array([ 0.80487805,  0.90243902,  0.85365854, ...,  0.95121951,
                  0.90243902,  1.        ])
In [204]: dm.shape
Out[204]: (2211,)

我想把这个数组转换成三角形矩阵或对称矩形矩阵。

 In [205]: reshapedDm = dm.trian_reshape(67, 67)

如何将trian_reshape实现为从1-D数组返回三角矩阵的函数？

Answer 1

>>> tri = np.zeros((67, 67))
>>> tri[np.triu_indices(67, 1)] = dm

有关详细信息，请参阅triu_indices的文档。要获得下三角矩阵，请使用np.tril_indices，并将偏移设置为-1而不是1。