networkx でラベル名を変更したい

Question

import networkx as nx

G = nx.barabasi_albert_graph(10, 1)
print(G.degree())
#=>DegreeView({0: 3, 1: 5, 2: 1, 3: 2, 4: 2, 5: 1, 6: 1, 7: 1, 8: 1, 9: 1})

networkｘの関数を使いグラフを作成しました。

ここでこの関数のラベル名を変更したいです。

理想は下のようになりたいです。

G.degree()
#=>DegreeView({a: 3, b: 5, c: 1, d: 2, e: 2, f: 1, g: 1, h: 1, i: 1, j: 1})

どのようにやればいいか見当がつきません。

Answer 1

degree() が返す関数はキーがノード、値が次数の辞書となっているので、ノードを 0, 1, ... から "a", "b", ... に変更したい場合は、nx.relabel_nodes でノード名を変更する必要があります。

import random
import numpy as np
import networkx as nx
import string

random.seed(42)  # 乱数シード固定

G = nx.barabasi_albert_graph(10, 1)
print(G.degree())
# [(0, 5), (1, 1), (2, 4), (3, 2), (4, 1), (5, 1), (6, 1), (7, 1), (8, 1), (9, 1)]

# ノードの名前を変更する。
mapping = {k: v for k, v in zip(G.nodes, string.ascii_lowercase)}
G = nx.relabel_nodes(G, mapping)
print(G.degree())
# [('a', 5), ('b', 1), ('c', 4), ('d', 2), ('e', 1), ('f', 1), ('g', 1), ('h', 1), ('i', 1), ('j', 1)]

参考文献

• networkx.relabel.relabel_nodes — NetworkX 2.4rc1.dev20190827025229 documentation

Comments

[hoshi1996]

よくわかりました
ありがとうございます。

Answer 2

よくわかりました。ありがとうございます。