如何使用 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()
输出
广告