python colormap自定义| LinearSegmentedColormap离散颜色列表转换为一个线性颜色表，并将得到的RGB值保存出来

前言

现在想自定义一个colormap，在网上搜索发现可以使用LinearSegmentedColormap将一个离散的颜色列表通过线性插值转换为一个指定颜色数目的colormap。之后就可以直接在python中绘图使用了。

import matplotlib as mpl
from pylab import *
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap,LinearSegmentedColormap

clist=['black','darkred','red']
# N 表示插值后的颜色数目
newcmp = LinearSegmentedColormap.from_list('chaos',clist,N=256)
aa = plt.imshow(np.random.randint(1,10,(5,5)),newcmp)
plt.colorbar(aa)

如何将此colormap保存下来用于其它的地方呢？

因为我主要是想通过python产生colormap，然后使用该colormap用于其它的地方绘图，因此得到底层产生的数据就十分有必要。增加以下几行代码就可以得到。

color_list = [[newcmp(i)[0],newcmp(i)[1],newcmp(i)[2] ]for i in range(newcmp.N)]
npList = np.array(color_list)
np.savetxt('black2read_cmp.txt',npList,fmt='%.3e')

如何直接将一列代表颜色的列表绘制成colorbar？

我们在拿到一列颜色值后，可能也会想直接将colorbar绘制处理，用于作图，放在文章中进行展示。只需要借助seaborn这个库就可以了。代码接着上文添加就行。绘制结果如下图。

import seaborn as sns
# 数据太多，可以做适当的降采样，以免得到的colorbar过于细长
sns.palplot(color_list[1:-1:2])

最后欢迎大家一起交流自定义colormap的各种方法，实现colormap自由。

完整代码

import matplotlib as mpl
from pylab import *
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap,LinearSegmentedColormap
import seaborn as sns
clist=['black','darkred','red']
# N 表示插值后的颜色数目
newcmp = LinearSegmentedColormap.from_list('chaos',clist,N=256)
aa = plt.imshow(np.random.randint(1,10,(5,5)),newcmp)
plt.colorbar(aa)

color_list = [[newcmp(i)[0],newcmp(i)[1],newcmp(i)[2] ]for i in range(newcmp.N)]
npList = np.array(color_list)
np.savetxt('Black2read2Write_cmp1.txt',npList,fmt='%.3e')

sns.palplot(color_list[1:-1:2])