如何用numpy矩阵构建newtworkx图？

import numpy as np
import networkx as nx

adj_matrix = np.array([[0, 1, 0, 0],
                      [1, 0, 1, 1],
                      [0, 1, 0, 1],
                      [0, 1, 1, 0]])

graph = nx.Graph()

num_nodes = adj_matrix.shape[0]
for i in range(num_nodes):
    graph.add_node(i)  # 添加节点
    for j in range(i+1, num_nodes):
        if adj_matrix[i][j] == 1:
            graph.add_edge(i, j)  # 添加边

print("图的节点：", graph.nodes())
print("图的边：", graph.edges())