自定义Matplotlib颜色图

debugcn 发表于 Dev

野餐眼镜

我想构造一个主要由三种颜色组成的色图，类似于色图“ brg”：

但是，我想在两个方面进行修改：

我想分别用蓝色，黑色和红色替换蓝色，红色和绿色。
我想扩展中间色的区域。作为一种规范，我目前正在使用matplotlibs'Normalize'，我想我可以选择非线性归一化，但是我宁愿只是将“ red（black）”添加到颜色表的中间并坚持线性归一化。

约翰·C

创建这种颜色图的最简单方法是通过LinearSegmentedColormap.from_list('', ['b', 'k', 'k', 'r'])。这个划分的颜色均匀，所以纯黑色占据1/3^次的范围内的。

该功能还允许通过形式（position，color）的元组来定位颜色，position范围从0左侧（底部）到1右侧（顶部）。下面是一个示例，其中黑色占了范围的10％。

import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap
from matplotlib.cm import ScalarMappable
import numpy as np

cmap1 = LinearSegmentedColormap.from_list('', ['b', 'k', 'k', 'r'])
plt.colorbar(ScalarMappable(cmap=cmap1), orientation='horizontal',
             ticks=np.linspace(0,1,11), label='pure black occupies 1/3rd')
width = 0.1
cmap2 = LinearSegmentedColormap.from_list('', [(0, 'b'), ((1 - width) / 2, 'k'), ( (1 + width) / 2, 'k'), (1, 'r')])
plt.colorbar(ScalarMappable(cmap=cmap2), orientation='horizontal',
             ticks=np.linspace(0,1,11), label=f'pure black occupies {width}')
plt.show()