在Seaborn FacetGrid中绘制多个DataFrame列

Question

我使用以下代码

import seaborn as sns

g = sns.FacetGrid(dataframe, col='A', hue='A')
g.map(plt.plot, 'X', 'Y1')
plt.show()

制作一个像这样的seaborn facet情节:

现在我想在y轴上用另一个变量添加另一行,称为Y2.结果应该类似于垂直堆叠由两个图获得的图

g = sns.FacetGrid(dataframe, col='A', hue='A')
g.map(plt.plot, 'X', 'Y1')
plt.show()

g = sns.FacetGrid(dataframe, col='A', hue='A')
g.map(plt.plot, 'X', 'Y2')
plt.show()

但在单个图中,没有重复的x轴和标题("A = <值>")并且没有创建新FacetGrid对象.

注意

g = sns.FacetGrid(dataframe, col='A', hue='A')
g.map(plt.plot, 'X', 'Y1')
g.map(plt.plot, 'X', 'Y2')
plt.show()

没有实现这一点,因为它导致Y1和Y2的曲线显示在A的每个值的相同子图中.

Answer 1

我使用以下代码创建了一个与您的数据集相匹配的合成数据集：

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

# Generate synthetic data
omega = np.linspace(0, 50)

A0s = [1., 18., 40., 100.]

dfs = []
for A0 in A0s:
    V_w_dr = np.sin(A0*omega)
    V_w_tr = np.cos(A0*omega)
    dfs.append(pd.DataFrame({'omega': omega,
                             'V_w_dr': V_w_dr,
                             'V_w_tr': V_w_tr,
                             'A0': A0}))
df = pd.concat(dfs, axis=0)

然后你就可以做你想做的了。感谢@mwaskom 在sharey='row', 和的评论中margin_titles=True：

dfm = df.melt(id_vars=['A0', 'omega'], value_vars=['V_w_dr', 'V_w_tr'])
g = sns.FacetGrid(dfm, col='A0', hue='A0', row='variable', sharey='row', margin_titles=True)
g.map(plt.plot, 'omega', 'value')

这导致

更新

• 在本次更新中，正确的方法是使用seaborn.relplot绘制 FacetGrid 的 。

sns.relplot(data=dfm, x='omega', y='value', col='A0', hue='A0', row='variable', kind='line')