python - Seaborn clustermap 颜色条调整

Question

在调整 seaborn.clustermap 中的颜色条时遇到一些问题。我正在尝试:

1. 水平调整颜色条
2. 更改颜色条边框颜色
3. 更改颜色条刻度长度

我检查了documentation对于 figure.colorbar 无济于事。

最少代码:

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

#data
corr = np.corrcoef(np.random.randn(10, 200))

#clustermap
kws = dict(cbar_kws=dict(label='Label', ticks=[0,0.50,1], orientation='horizontal'), figsize=(6, 6))
sns.clustermap(corr, cmap="Blues", xticklabels=False, yticklabels=False, **kws)
plt.show()

Answer 1

您可以通过g.ax_cbar使用颜色条抓取子图。然后您可以更改其位置、标题、书脊和刻度长度:

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

corr = np.corrcoef(np.random.randn(10, 200))

kws = dict(cbar_kws=dict(ticks=[0, 0.50, 1], orientation='horizontal'), figsize=(6, 6))
g = sns.clustermap(corr, cmap="Blues", xticklabels=False, yticklabels=False, **kws)
x0, _y0, _w, _h = g.cbar_pos
g.ax_cbar.set_position([x0, 0.9, g.ax_row_dendrogram.get_position().width, 0.02])
g.ax_cbar.set_title('colorbar title')
g.ax_cbar.tick_params(axis='x', length=10)
for spine in g.ax_cbar.spines:
    g.ax_cbar.spines[spine].set_color('crimson')
    g.ax_cbar.spines[spine].set_linewidth(2)
plt.show()