NetworkX 最佳实践在迭代边缘时获取边缘属性值

Question

给定一个边列表（或生成器）。识别具有特定属性值的边的最易读方法是什么？例如，所有带有"foo"edge_type"的边缘？

目前，我的代码如下所示：

for edge in nx_graph.out_edges(my_node):
   edge_type = nx_graph[edge[0]][edge[1]]['edge_type']
   if edge_type == 'foo':
       ...

由于括号很多，这不太容易阅读......

一个稍微更具可读性的方法：

for edge in G.edges_iter(data=True):
    if edge[2]['edge_type']=='foo':
        ...

然而，它仍然不是很清楚（尤其是[2]）。另外，我不确定如何将其与out_edges()一起使用

Answer 1

这是一个选项

for edge in ((u,v,data) for u,v,data in G.edges_iter(data=True) if data['edge_type']=='foo'): 
    ...