networkxでグラフを描く

= python の networkx を使ってグラフを作ってみたのでメモ

# -*- encoding:utf-8 -*-

import networkx
import pylab
from matplotlib import font_manager
from itertools import combinations
from random import randint

# ノードをkey、エッジをはるノードのlistをvalueとするdict
vector = {}
persons = [u"田中", u"鈴木", u"山田", u"木村", u"吉岡"]
edge_labels = {}

for person in persons:
    # defaultdict(list)ではなく、ノードを作成するためにこうする
    vector[person] = []

for man_pair in combinations(persons, 2):
    man1, man2 = man_pair
    # 適当にエッジに値を付ける
    r = randint(1, 10)
    if r % 2:
        continue
    else:
        vector[man1].append(man2)
        edge_labels[(man1, man2)] = r

graph = networkx.Graph(vector)  # 無向グラフ
# graph = network.DiGraph(vector)  # 有向グラフ (to_undirectedで無向グラフに変換可）
pylab.figure(figsize=(3, 4))  # 横3inch 縦4inchのサイズにする
pos = networkx.spring_layout(graph)  # いい感じにplotする
# pos = networkx.random_layout(graph)  とでもすれば高速にplot出来る



# フォントを変更する（font_pathは適宜変更する）
font_path = "/usr/share/fonts/japanese/TrueType/sazanami-gothic.ttf"
font_prop = font_manager.FontProperties(fname=font_path)
networkx.set_fontproperties(font_prop)

# 見た目をいじる
networkx.draw_networkx_nodes(graph, pos, node_size=100, node_color="w")
networkx.draw_networkx_edges(graph, pos, width=1)
networkx.draw_networkx_edge_labels(graph, pos, edge_labels=edge_labels)
networkx.draw_networkx_labels(graph, pos, font_size=16, font_color="r")

pylab.xticks([])
pylab.yticks([])

pylab.show()
pylab.savefig("graph_networkx.png")

ここで、networkx のバージョンが 1.5 以上でなければ日本語は表示できません。 おそらく、

ValueError: matplotlib display text must have all code points < 128 or use Unicode strings

というエラーが出るかと思います。 日本語を表示させるためにはここのパッチを networkx/drawing/nx_pylab.py に適用してください

##結果 こんな感じに

*** 参考 http://d.hatena.ne.jp/nishiohirokazu/20111121/1321849806 http://antibayesian.hateblo.jp/entry/20110828/1314491180

from: https://qiita.com/petitviolet/items/d66cd7db5b35aec91c35