python – 在draw_networkx和matplotlib中设置颜色

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 在draw_networkx和matplotlib中设置颜色，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含2002字，纯文字阅读大概需要3分钟。

内容图文

我正在绘制两种类型的图形,我想要对齐颜色：

> networkx.Graph中的节点颜色,和
>普通情节中pyplot.hlines的颜色.

这两个数字都依赖于一个字典,其中node_names作为键,int作为值,以及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_network如何使用node_color和cmap我有什么遗漏？

解决方法:

您需要将值映射到[0,1]色彩映射范围.为您的hline绘图明确地执行此操作 – networkx的maptlotlib界面将为您执行此操作,您可以指定最小值和最大值.请注意,对于调用networkx.draw_networkx,nodelist order和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()

内容总结

以上是互联网集市为您收集整理的python – 在draw_networkx和matplotlib中设置颜色全部内容，希望文章能够帮你解决python – 在draw_networkx和matplotlib中设置颜色所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端