python - 将图中的所有节点重命名为数字序列

Question

我正在为 Python 使用 NetworkX 图形库。在我的程序中的某个时刻，我想将我的 nodeID“合并”成一个数字序列。这是我天真的方法：

start = 1 # could be anything
for i, n in enumerate(g.nodes()):
    if i+start == n:
        continue
    g.add_node(i+start, attr_dict=g.node[n])
    g.add_edges_from([(i+start, v, g[n][v]) for v in g.neighbors(n)])
    g.remove_node(n)

有没有比所有邻居的详尽副本更快的方法？例如，我试过g[i+start] = g[n]了，但这是被禁止的。

谢谢！

Answer 1

这行得通吗？

http://networkx.github.io/documentation/latest/reference/generated/networkx.relabel.convert_node_labels_to_integers.html

import networkx as nx
G = nx.Graph()
G.add_node(1)
G.add_nodes_from('spam')
print G.nodes()

返回：

['a', 1, 's', 'm', 'p']

现在：

start = 1
G = nx.convert_node_labels_to_integers(G,first_label=start)
print G.nodes()

返回：

[1, 2, 3, 4, 5]

Answer 2

如果您的兴趣仍然相关，则networkx.relabel_nodes()可以使用映射字典。