Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

networkx와 Matplotlib를 사용하여 다자간 그래프를 만드는 방법은 무엇입니까?

<시간/>

networkx에서 다자간 그래프를 만들려면 우리는 다음 단계를 수행할 수 있습니다 -

  • Figure 크기를 설정하고 서브플롯 사이와 주변의 패딩을 조정합니다.

  • 하위 집합 크기 및 색상 목록을 만듭니다.

  • 다층 그래프 개체를 반환할 수 있는 다층 그래프에 대한 메서드를 정의합니다.

  • 노드의 색상을 설정합니다.

  • 노드를 직선 레이어에 배치합니다.

  • 그래프 G 그리기 Matplotlib와 함께.

  • 동일한 축 속성을 설정합니다.

  • 그림을 표시하려면 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()

출력

networkx와 Matplotlib를 사용하여 다자간 그래프를 만드는 방법은 무엇입니까?