在Seaborn FacetGrid中打印多个数据帧列

Question

我使用以下代码

import seaborn as sns

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

要制作这样的海生小面图：

现在我想在这个图中添加另一行，在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=<；value>；”），也没有创建新的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
import numpy
import seaborn as sns
import matplotlib.pyplot as plt

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

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

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

然后你可以做你想做的事情（感谢@mwaskom在)sharey='row', margin_titles=True的评论中）：

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

这将导致