两色以上的海运diverging_palette

Question

我试图用Seaborn绘制一个相关矩阵，但我想突出红色和绿色中间值的正负极值。在我可以找到的所有示例中，相关矩阵都是用diverging_palette绘制的，但这只允许您为光谱的末尾选择两种颜色，并为中间值选择光(白色)或暗(黑色)颜色。在StackOverflow和其他网站上搜索之后，我无法找到解决方案，所以我发布了我找到的解决方案。

以下是Seaborn的例子：

correlations.html https://seaborn.pydata.org/generated/seaborn.heatmap.html

下面是代码来生成下面的图来说明这个问题。我要找的是绿色是0值，红色是正负值。

import pandas as pd
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

np.random.seed(5)
df = pd.DataFrame(np.random.randn(5,5))

# Generate a custom diverging colormap
cmap = sns.diverging_palette(133, 10, as_cmap=True)

with sns.axes_style("white"):
    ax = sns.heatmap(df, annot=True, fmt='.2f', cmap=cmap, vmin=-0.99, vmax=.99, center=0.00,
                square=True, linewidths=.5, annot_kws={"size": 8}, cbar_kws={"shrink": .5})
plt.show()

​

​

Answer 1

您可以使用LinearSegmentedColormap从一个颜色列表创建一个颜色映射，该列表允许您通过对颜色转换进行更大的控制来实现diverging_pallette的效果。

将此代码插入上面的示例将产生我正在搜索的效果：

from matplotlib.colors import LinearSegmentedColormap
cmap = LinearSegmentedColormap.from_list(
    name='test', 
    colors=['red','white','green','white','red']
)

​

​