简化networkx节点标签

Question

%matplotlib inline
import networkx as nx
import matplotlib.pyplot as plt

G = nx.Graph()
G.add_node('abc@gmail.com')
nx.draw(G, with_labels=True)
plt.show()

输出图为

我想要的是

我在 csv 文件中有数千条从 person@email.com 到 another@email.com 的电子邮件记录，我使用 G.add_node(email_address) 和 G.add_edge(from, to) 构建 G。我想将整个电子邮件地址保留在 Graph G 中，但是以简化的字符串显示。

Answer 1

networkx 有一个名为 relabel_nodes 的方法，它接受一个图形 (G)、一个 mapping（重新标记规则）并返回一个新图形 (new_G)，其中节点重新标记。

也就是说，在你的情况下：

import networkx as nx
import matplotlib.pyplot as plt

G = nx.Graph()
G.add_node('abc@gmail.com')
mapping = {
   'abc@gmail.com': 'abc'
}
relabeled_G = nx.relabel_nodes(G,mapping)
nx.draw(relabeled_G, with_labels=True)
plt.show()

这样您就可以保持G 完整并具有简化的标签。

您可以选择就地修改标签，而无需新副本，在这种情况下，您只需调用G = nx.relabel_nodes(G, mapping, copy=False)

如果你事先不知道邮箱地址，你可以传递relabel_nodes一个函数，像这样：

G = nx.relabel_nodes(G, lambda email: email.split("@")[0], copy=False)