一个colorbar用于subplot中的seaborn heatmaps

Question

这是一个显示每个子图的颜色条的示例:

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

df = pd.DataFrame(np.random.random((10,10,)))

fig,axn = plt.subplots(2, 2, sharex=True, sharey=True)

for ax in axn.flat:
    sns.heatmap(df, ax=ax)

如何删除每个子图的颜色条？我想只有一个垂直或水平方向的颜色条.我知道我可以访问每个颜色条轴fig.get_axes()[:-4],但是如何从图中完全删除它们呢？我不认为在调用热图时可以选择不绘制颜色条.

Answer 1

该cbar参数控制是否应添加颜色条,cbar_ax参数可以选择指定颜色条应该到达的轴.所以,你可以这样做:

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

df = pd.DataFrame(np.random.random((10,10,)))

fig, axn = plt.subplots(2, 2, sharex=True, sharey=True)
cbar_ax = fig.add_axes([.91, .3, .03, .4])

for i, ax in enumerate(axn.flat):
    sns.heatmap(df, ax=ax,
                cbar=i == 0,
                vmin=0, vmax=1,
                cbar_ax=None if i else cbar_ax)

fig.tight_layout(rect=[0, 0, .9, 1])

(你会tight_layout在这里得到一个警告,但它实际上是正确的,因为我们cbar_ax明确地放置了.如果你不喜欢看到警告,你也可以tight_layout在绘图之前打电话,但它不会那么紧张).

Answer 2

实际上没有必要将cbar_ax前 3 个子图设置为 none。您可以cbar_ax=cbar_ax对所有 4 个子图进行设置，它只会在完全相同的位置绘制颜色条 4 次，这根本不会影响外观。

这对于使用 FacetGrid 的人来说效果更好，例如给定一个数据框df：

def draw_heatmap(*args, **kwargs):
    data = kwargs.pop('data')
    d = data.pivot(index=args[1], columns=args[0], values=args[2])
    sns.heatmap(d, **kwargs)

g = sns.FacetGrid(df, col='col_name', col_wrap=2, margin_titles=True, sharey=True)

cbar_ax = g.fig.add_axes([.91, .15, .03, .7])
g = g.map_dataframe(draw_heatmap, 'col_col', 'index_col', 'val_col', annot=True, 
                    cmap='Spectral', cbar_ax=cbar_ax, cbar_kws={'label': 'color_bar_label'})