行列をグラフ化してネットワーク図を作成したい

Question

前提・実現したいこと

Python初心者です。
Pythonでresponseを用いてjson形式で情報を取得し、pandasを利用してそれをcsvファイルに変換した後、
csvファイルを読み込んで隣接行列を作成し、出力するプログラムを作成しました。

そこで、行列をグラフ化してネットワーク図？（可視化）を出力したいと考えていますが、方法が思い浮かばない状況です。
networkxというライブラリを使うとよさそうなのですが、使い方が分かりません。

どなたか、教えて頂けないでしょうか。参考にさせて頂きたいです。
以下に行列を出力するまでのソースコードを提示しておきます。
宜しくお願い致します。

補足情報

Ubuntu 20.04 LTS/python3

Answer 1

元データのカタチが不明ですが、以下のような感じで描画できます。
networkx.convert_matrix.from_pandas_edgelistを使うと、隣接行列を作らずに直接ノードリストからグラフを生成できます。

import pandas as pd
from io import StringIO
import matplotlib.pyplot as plt
import networkx as nx

# テストデータ
s = """node1,node2
a,b
a,c
b,c"""

df = pd.read_csv(StringIO(s))
G = nx.from_pandas_edgelist(df, 'node1', 'node2')
nx.draw(G, with_labels = True)
plt.show()

# 参考：隣接行列から生成
df = pd.read_csv(StringIO(s))
df = df.drop_duplicates()
df = pd.crosstab(df.node1, df.node2)
idx = df.columns.union(df.index)
df = df.reindex(index = idx, columns=idx, fill_value=0)
a = df.values
G = nx.from_numpy_matrix(a)
nx.draw(G)
plt.show()

Comments

[eve]

ありがとうございます。想定していた図を出力できました。
参考になりました。