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 实现为从一维数组返回三角矩阵的函数？

Answer 1

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

详情请参阅doc for triu_indices。要获得下三角矩阵，请使用np.tril_indices 并将偏移量设置为-1 而不是1。