从属性中选择具有networkx图的节点和边

Question

我刚刚开始在networkx中做图表，我想及时跟踪图表的演变：它是如何变化的，图表中指定时间点的节点/边缘是什么。

这是我的代码：

import networkx as nx
import matplotlib.pyplot as plt
G=nx.Graph()
G.add_node(1,id=1000,since='December 2008')
G.add_node(2,id=2000,since='December 2008')
G.add_node(3,id=3000,since='January 2010')
G.add_node(4,id=2000,since='December 2016')
G.add_edge(1,2,since='December 2008')
G.add_edge(1,3,since='February 2010')
G.add_edge(2,3,since='March 2014')
G.add_edge(2,4,since='April 2017')
nx.draw_spectral(G,with_labels=True,node_size=3000)
plt.show()

这显示了包含所有节点和边的图。

所以，我的问题是：

如何设计基于时间的过滤器，该过滤器将在时间t仅提取图表G图上的相关节点/边缘，例如＆＃39; 2014年7月＆＃39;。完成后，如何使用matplotlib更新图形？

提前感谢您的帮助

Answer 1

您可以使用带有G.nodes()方法的列表理解条件选择节点：

selected_nodes = [n for n,v in G.nodes(data=True) if v['since'] == 'December 2008']  
print (selected_nodes)

出：[1, 2]

要使用G.edges_iter或G.edges方法选择边缘：

selected_edges = [(u,v) for u,v,e in G.edges(data=True) if e['since'] == 'December 2008']
print (selected_edges)

出：[(1, 2)]

要绘制所选节点，请致电G.subgraph()

H = G.subgraph(selected_nodes)
nx.draw(H,with_labels=True,node_size=3000)

要使用属性绘制选定边，您可以构建新图：

H = nx.Graph(((u, v, e) for u,v,e in G.edges_iter(data=True) if e['since'] == 'December 2008'))
nx.draw(H,with_labels=True,node_size=3000)
plt.show()