Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Vẽ đồ thị mạng (network graph) với NetworkX và Matplotlib

Để vẽ đồ thị mạng với NetworkX và Matplotlib, ta dùng plt.show() sau khi vẽ.

Các bước thực hiện

  • Thiết lập kích thước hình và khoảng đệm.
  • Tạo DataFrame với khóa from và to.
  • Tạo đồ thị chứa edgelist từ DataFrame.
  • Vẽ đồ thị bằng phương thức draw() với các thuộc tính node.
  • Hiển thị hình bằng show().

Ví dụ

import pandas as pd
import networkx as nx
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

df = pd.DataFrame({'from': ['A', 'B', 'C', 'A'], 'to': ['D', 'A', 'E', 'C']})
G = nx.from_pandas_edgelist(df, 'from', 'to')
nx.draw(G, with_labels=True, node_size=100, alpha=1, linewidths=10)
plt.show()

Kết quả

Vẽ đồ thị mạng (network graph) với NetworkX và Matplotlib