如何使用 networkx 和 Matplotlib 製作多部分圖?


要製作networkx 中的多部分圖,我們可以執行以下步驟 −

  • 設定圖表大小並調整子圖表內外的填充。

  • 建立子集大小和顏色的列表。

  • 定義多層圖的方法,可以返回一個多層圖物件。

  • 設定節點的顏色。

  • 將節點定位在直線層中。

  • 使用 Matplotlib 繪製圖G

  • 設定相等的軸屬性。

  • 使用show()方法顯示圖表。

示例

import itertools
import matplotlib.pyplot as plt
import networkx as nx

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

subset_sizes = [5, 5, 4, 3, 2, 4, 4, 3]
subset_color = [
   "gold",
   "violet",
   "violet",
   "violet",
   "violet",
   "limegreen",
   "limegreen",
   "darkorange",
]

def multilayered_graph(*subset_sizes):
   extents = nx.utils.pairwise(itertools.accumulate((0,) + subset_sizes))
   layers = [range(start, end) for start, end in extents]
   G = nx.Graph()
   for (i, layer) in enumerate(layers):
      G.add_nodes_from(layer, layer=i)
   for layer1, layer2 in nx.utils.pairwise(layers):
      G.add_edges_from(itertools.product(layer1, layer2))
   return G

G = multilayered_graph(*subset_sizes)
color = [subset_color[data["layer"]] for v, data in G.nodes(data=True)]
pos = nx.multipartite_layout(G, subset_key="layer")
nx.draw(G, pos, node_color=color, with_labels=False)

plt.axis("equal")

plt.show()

輸出

更新於: 2021-08-10

1k+ 瀏覽

開啟你的事業

完成課程並獲得認證

開始
廣告
© . All rights reserved.