在python矩阵中将上三角复制到下三角

2024-04-26 09:43:54 发布

您现在位置:Python中文网/ 问答频道 /正文

       iluropoda_melanoleuca  bos_taurus  callithrix_jacchus  canis_familiaris
ailuropoda_melanoleuca     0        84.6                97.4                44
bos_taurus                 0           0                97.4              84.6
callithrix_jacchus         0           0                   0              97.4
canis_familiaris           0           0                   0                 0

这是我拥有的python矩阵的一个简短版本。我有上三角的信息。是否有一个简单的函数可以将矩阵的上三角复制到下三角?


Tags: 函数版本信息矩阵bostaurusailuropodajacchus
3条回答

如果我正确地理解了这个问题,我相信这是可行的

for i in range(num_rows):
    for j in range(i, num_cols):
        matrix[j][i] = matrix[i][j]

要在NumPy中执行此操作,无需使用双循环,您可以使用^{}

>>> i_lower = np.tril_indices(n, -1)
>>> matrix[i_lower] = matrix.T[i_lower]  # make the matrix symmetric

请注意,不要尝试混合使用tril_indices^{},因为它们都使用行主索引,即,这不起作用:

>>> i_upper = np.triu_indices(n, 1)
>>> i_lower = np.tril_indices(n, -1)
>>> matrix[i_lower] = matrix[i_upper]  # make the matrix symmetric
>>> np.allclose(dist.T, dist)
False

最简单、最快(无循环)的方法如下:

import numpy as np

X= np.array([[0., 2., 3.],
             [0., 0., 6.],
             [0., 0., 0.]])

X = X + X.T - np.diag(np.diag(X))
print(X)

#array([[0., 2., 3.],
#       [2., 0., 6.],
#       [3., 6., 0.]])

相关问题 更多 >