Networkx:如何选择具有给定键值的所有边？

import networkx as nx

# 创建一个有向图
G = nx.DiGraph()

# 添加边和属性
G.add_edge('A', 'B', weight=4)
G.add_edge('B', 'C', weight=2)
G.add_edge('C', 'D', weight=1)
G.add_edge('A', 'D', weight=3)

# 选择具有给定键值的所有边
selected_edges = [(u, v, d) for u, v, d in G.edges(data=True) if 'weight' in d and d['weight'] == 2]

# 打印选择的边
for u, v, d in selected_edges:
    print(f"边 ({u}, {v}) 的权重为 {d['weight']}")