使用seaborn heatmap之后,我有几个颜色条作为轴实例,但不知道如何在结果颜色条周围绘制黑色轮廓线。
可以使用
我已经尝试过
这将产生以下结果:
我可以直接修改seaborn matrix.py文件,但我希望有一种matplotlib的方法。
可以使用
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的方法。