Python 在draw_networkx和matplotlib中设置颜色

我正在绘制两种类型的图形，我希望将其颜色对齐：

networkx.Graph中节点的颜色，以及

常规打印中pyplot.hline的颜色

这两个图都依赖于一个字典，其中节点名称作为键，整数作为值，以及

matplotlib.colors.LinearSegmentedColormap

cmap。例如：

import matplotlib.pyplot as plt
Dict = {"Alice": 0, "Bob": 1, "Carol": 2}
cmap = plt.cm.Accent

这使我能够为每个名称获得唯一的颜色。对于此示例：

for key, value in Dict.iteritems():
    print key, value, cmap(value)

给我

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\u networkx

如何使用

node\u color

和

cmap

，我缺少什么？

您需要将值映射到[0,1]颜色映射范围。为您的hline绘图明确执行此操作-networkx的maptlotlib接口将为您执行此操作，您可以指定最小值和最大值。请注意，对于调用

networkx.draw\u networkx

而言，节点列表顺序和节点颜色顺序必须相同

import networkx as nx
import matplotlib.pyplot as plt
import matplotlib as mpl
Dict = {"Alice": 0, "Bob": 1, "Carol": 2}
cmap = plt.cm.Accent
norm = mpl.colors.Normalize(vmin=0, vmax=2)
m = plt.cm.ScalarMappable(norm=norm, cmap=cmap)
for key, value in Dict.iteritems():
    print key, value, m.to_rgba(value)
plt.hlines(1, 1, 5, m.to_rgba(Dict["Alice"]), linewidth=50)
plt.hlines(2, 1, 5, m.to_rgba(Dict["Bob"]),   linewidth=50)
plt.hlines(3, 1, 5, m.to_rgba(Dict["Carol"]), linewidth=50)
G = nx.Graph()
G.add_nodes_from(Dict.keys())

nodelist,node_color = zip(*Dict.items())
nx.draw_networkx(G, nodelist=nodelist, node_size=1000, node_color=node_color,vmin=0.0,vmax=2.0, cmap=cmap)
plt.show()