首页 > 解决方案 > 基于 facet 在 Seaborn FacetGrid 中自定义标题

问题描述

想做一个分面网格,变量名作为列,部门作为行,每个小图都是y=value和x=date的散点图

我的数据是这样的:

import pandas as pd import numpy as np import seaborn as sns import matplotlib.pyplot as plt from datetime import datetime import matplotlib.dates as mdates import random

datelist = pd.date_range(start="march 1 2020", end="may 20 2020", freq="w").tolist()
varlist = ["x", "y", "z", "x", "y", "z", "x", "y", "z", "x", "y", "z"]
deptlist = ["a", "a", "b", "a", "a", "b", "a", "a", "b", "a", "a", "b"]
vallist =  random.sample(range(10, 30), 12)
df = pd.DataFrame({'date': datelist, 'value': vallist, 'variable': varlist, 'department': deptlist})

我想像这样创建一个 Seaborn Facetgrid:

g = sns.FacetGrid(df, row="department", col="variable", sharey='row')

我想更改 Y 轴标签以包含部门名称,例如“部门 A”等。但根据我使用的数据,特定部门可能会有所不同。所以我想用我从 g 中提取的东西来参数化 Y 轴标签的设置,它告诉我映射到某个行的特定变量。我一直在四处挖掘,虽然我确定它存在但我找不到它。

标签: pythonseaborn

解决方案


您可以使用set_ylabel()set_xlabel更改特定方面的标签。因此,在您提供的示例中,department = a | variable = x构面将是 [0,0],下一个标签位于 [1,0]。

所以我们可以试试这个,使用 row_order 输入来确保我们有相同的顺序:

rowOrder = ['a','b']
g = sns.FacetGrid(df, row="department", col="variable", sharey='row',row_order=rowOrder)
g = g.map(plt.scatter, "value", "value")
for i in range(len(rowOrder)):
    g.axes[i,0].set_ylabel(rowOrder[i])

在此处输入图像描述


推荐阅读