python - 如何为 Seaborn Facet Plot 添加标题

Question

如何为这个 Seaborne 图添加标题？让我们给它起一个标题“I AM A TITLE”。

tips = sns.load_dataset("tips")
g = sns.FacetGrid(tips, col="sex", row="smoker", margin_titles=True)
g.map(sns.plt.scatter, "total_bill", "tip")

Answer 1

稍微更新，seaborn 0.11.1:

Seaborn 的 relplot 函数创建了一个 FacetGrid 并为每个子图提供了自己的解释性标题。您可以在整个内容上添加标题:

import seaborn as sns
tips = sns.load_dataset('tips')

rp = sns.relplot(data=tips, x='total_bill', y='tip',
                 col='sex', row='smoker',
                 kind='scatter')
# rp is a FacetGrid; 
# relplot is a nice organized way to use it

rp.fig.subplots_adjust(top=0.9) # adjust the Figure in rp
rp.fig.suptitle('ONE TITLE FOR ALL')

如果您直接创建 FacetGrid，就像在原始示例中一样，它会自动添加列和行标签，而不是单独的子图标题。我们仍然可以为整个事物添加标题:

from matplotlib.pyplot import scatter as plt_scatter
g = sns.FacetGrid(tips, col='sex', row='smoker', 
                  margin_titles=True)
g.map(plt_scatter, 'total_bill', 'tip')
g.fig.subplots_adjust(top=0.9)
g.fig.suptitle('TITLE!')

FacetGrid 对象是用 matplotlib Figure 对象构建的，因此我们可以使用 subplots_adjust、suptitle，这些在 matplotlib 中通常很熟悉。