matplotlib 使用你自己想要的color map

最新推荐文章于 2025-07-16 10:00:00 发布

原创最新推荐文章于 2025-07-16 10:00:00 发布 · 1.8w 阅读

33 ·

CC 4.0 BY-SA版权

文章标签：

#Python

本文介绍如何在Matplotlib中创建自定义的红-绿-蓝渐变颜色映射，通过LinearSegmentedColormap从List方法实现，并注册到Matplotlib中供绘图使用。

最近使用matplotlib画图，想要使用一个三段的红-绿-蓝映射，并且带有一定梯度，就像这样：

不过，发现matplotlib中虽然带有很多映射方式，但是只有brg方式，没有rgb方式

就像这样：

nx.draw_networkx(G,pos, edge_color = edge_length, edge_cmap = plt.get_cmap('brg'))

那么怎么办呢？如果直接将[0,1]等分三段，分别是纯红、纯绿、纯蓝，是非常难看的。

于是将官网例子http://scipy.github.io/old-wiki/pages/Cookbook/Matplotlib/Show_colormaps

（这是非常好的例子，各种color map 方式都有）

修改了一下：

import matplotlib
import matplotlib.colors as col
import matplotlib.cm as cm
import matplotlib.pyplot as plt
import numpy as np

# example 2: use the "fromList() method
startcolor = '#ff0000'   #红色，读者可以自行修改
midcolor = '#00ff00'     #绿色，读者可以自行修改
endcolor = '#0000ff'          #蓝色，读者可以自行修改
cmap2 = col.LinearSegmentedColormap.from_list('own2',[startcolor,midcolor,endcolor])
# extra arguments are N=256, gamma=1.0
cm.register_cmap(cmap=cmap2)
# we can skip name here as it was already defined

当register_cmap执行完毕后，调用该color map 的名称 'own2'，即可。

cm.get_cmap('own2')

就获得了文章开头的红-绿-蓝有过渡的颜色映射。函数自动使用插值计算0-1之间的颜色。