在sea born / matplotlib中绘制轮廓圆形颜色条

Question

使用seaborn热图后，我有几个颜色条作为轴实例，但不知道如何在结果颜色条周围绘制黑色轮廓。

我可以使用axhline和axvline打开x轴和y轴，但是如何绘制右刺和顶刺以完成框。

我尝试了set_visible，set_linewidth，set_color，但无济于事。

import numpy as np 
from pandas import DataFrame
import seaborn as sns
import matplotlib.pyplot as plt


Index= ['aaa', 'bbb', 'ccc', 'ddd', 'eee']
Cols = ['A', 'B', 'C', 'D']
df = DataFrame(abs(np.random.randn(5, 4)), index=Index, columns=Cols)


figure = plt.figure(figsize=(12,12))
grid_spec = plt.GridSpec(1,4, width_ratios=[5,1,5,1], height_ratios=[1,1], wspace=0.6, hspace=0.3)

heatmap1ax = figure.add_subplot(grid_spec[0])
cbar1ax = figure.add_subplot(grid_spec[1])
heatmap2ax = figure.add_subplot(grid_spec[2])
cbar2ax = figure.add_subplot(grid_spec[3])

sns.set_style("white", {'axes.linewidth': 1, 'grid.color': 'black', 'xtick.major.size': 10, 'ytick.major.size': 0}) 
sns.heatmap(df, cmap='cool', ax=heatmap1ax, cbar_ax=cbar1ax)
sns.heatmap(df[['A']], cmap='cool', ax=heatmap2ax, cbar_ax=cbar2ax)

for axis in ['top','bottom','left','right']:
    heatmap1ax.spines[axis].set_visible(True)
    heatmap1ax.spines[axis].set_color('black')
    heatmap2ax.spines[axis].set_visible(True)
    heatmap2ax.spines[axis].set_color('black')    

plt.show()

这会产生以下结果：

但是如何在彩条中添加类似的轮廓？

我可以直接修改seaborn matrix.py文件，但我希望有一个matplotlib路由。

Answer 1

我认为您要查找的命令是set_frame_on。来自matplotlib's 'axes' API：

  

set_frame_on（b）中

     

设置是否绘制轴矩形补丁

     

ACCEPTS：[True |错]

所以，添加到您的代码

cbar1ax.set_frame_on(True)
cbar2ax.set_frame_on(True)

以你想要的方式为你的情节中的每个颜色条添加一个框架（如果我理解正确的话）。