在 draw_networkx 和 matplotlib 中设置颜色
Setting colors in draw_networkx and in matplotlib
我正在画两种类型的图形,我想为它们调整颜色:
- networkx.Graph和
中节点的颜色
- 常规绘图中 pyplot.hlines 的颜色。
这两个数字都依赖于以 node_names 作为键、整数作为值和 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 种明显不同的颜色,但对于 hlines,它们几乎无法区分。
关于 nx.draw_networkx
如何使用 node_color
和 cmap
我错过了什么?
您需要将您的值映射到 [0,1] 色图范围内。为您的 hline 图明确地执行此操作 - networkx 的 maptlotlib 接口将为您执行此操作,您可以指定最小值和最大值。请注意,对于调用 networkx.draw_networkx
,节点列表顺序和 node_color 顺序必须相同。
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()
我正在画两种类型的图形,我想为它们调整颜色:
- networkx.Graph和 中节点的颜色
- 常规绘图中 pyplot.hlines 的颜色。
这两个数字都依赖于以 node_names 作为键、整数作为值和 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 种明显不同的颜色,但对于 hlines,它们几乎无法区分。
关于 nx.draw_networkx
如何使用 node_color
和 cmap
我错过了什么?
您需要将您的值映射到 [0,1] 色图范围内。为您的 hline 图明确地执行此操作 - networkx 的 maptlotlib 接口将为您执行此操作,您可以指定最小值和最大值。请注意,对于调用 networkx.draw_networkx
,节点列表顺序和 node_color 顺序必须相同。
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()