Каков наиболее эффективный способ переместить/переименовать node в NetworkX?

Я использую библиотеку графиков NetworkX для Python. В какой-то момент в моей программе я хотел бы "консолидировать" мои идентификаторы узлов в последовательности чисел. Вот мой наивный подход:

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], но это запрещено.

Спасибо!

Ответ 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]

Ответ 2

Если ваш интерес по-прежнему имеет значение, networkx.relabel_nodes() принимает словарь сопоставления.