如何在python中从矩阵传递到Dic无向图rapresentation python？

def matrix_to_graph(matrix):
    graph = {}
    rows = len(matrix)
    cols = len(matrix[0])
    
    for i in range(rows):
        for j in range(cols):
            if matrix[i][j] != 0:
                if i not in graph:
                    graph[i] = []
                graph[i].append((j, matrix[i][j]))
                
                if j not in graph:
                    graph[j] = []
                graph[j].append((i, matrix[i][j]))
    
    return graph

# 示例矩阵
matrix = [
    [0, 1, 0, 1],
    [1, 0, 1, 0],
    [0, 1, 0, 1],
    [1, 0, 1, 0]
]

graph = matrix_to_graph(matrix)
print(graph)

{0: [(1, 1), (3, 1)], 1: [(0, 1), (2, 1)], 2: [(1, 1), (3, 1)], 3: [(0, 1), (2, 1)]}