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

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