我画了两种类型的图形,我想把它们的颜色对齐:
这两个图都依赖于一个字典,其中节点名作为键,int作为值,以及一个matplotlib.colors.LinearSegmentedColormap
cmap。如下所示:
import matplotlib.pyplot as plt
Dict = {"Alice": 0, "Bob": 1, "Carol": 2}
cmap = plt.cm.Accent
这样我就可以为每个名字取一种独特的颜色。对于本例:
^{pr2}$给了我
Bob 1 (0.50482122313742539, 0.78532873251858881, 0.50718954287323292, 1.0)
Alice 0 (0.49803921580314636, 0.78823530673980713, 0.49803921580314636, 1.0)
Carol 2 (0.51160323047170453, 0.7824221582973705, 0.5163398699433196, 1.0)
其用途如下:
plt.hlines(1, 1, 5, cmap(Dict["Alice"]))
plt.hlines(2, 1, 5, cmap(Dict["Bob"]))
plt.hlines(3, 1, 5, cmap(Dict["Carol"]))
但是,这些值与绘制以下网络时得到的结果完全不匹配:
G = nx.Graph()
G.add_nodes_from(Dict.keys())
nx.draw_networkx(G, nodelist=Dict.keys(), node_color=range(3), \
cmap=plt.cm.Accent)
对于网络的情节,我得到了3种明显不同的颜色,但对于hline,它们几乎无法区分。在
关于nx.draw_networkx
如何使用node_color
和{
您需要将值映射到[0,1]colormap范围。为您的hline绘图显式执行此操作-networkx到maptlotlib的接口将为您完成此操作,您可以指定最小值和最大值。请注意,对于调用
networkx.draw_networkx
,nodelist顺序和node\u颜色顺序必须相同。在相关问题 更多 >
编程相关推荐