Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีปรับรูปแบบ networkx graph ใน Python


ปรับโครงสร้างกราฟ networkx

สร้างกราฟจาก Pandas DataFrame edge list แล้ววาดด้วย networkx

ขั้นตอน

  1. สร้าง DataFrame ด้วย Pandas
  2. สร้างกราฟจาก edge list ด้วย from_pandas_edgelist()
  3. วาดกราฟ G ด้วย Matplotlib ปรับโครงสร้างได้โดยแก้ไขคีย์ "from" และ "to"
  4. แสดงด้วย show()

ตัวอย่างโค้ด

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

plt.rcParams["figure.figsize"] = [7.00, 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=150, alpha=0.5, linewidths=40)
plt.show()

ผลลัพธ์

วิธีปรับรูปแบบ networkx graph ใน Python