python - 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。