Seaborn图在同一散点图上的两个数据集

时间:2018-08-07 18:01:40

标签: python python-3.x pandas matplotlib seaborn

我在Pandas Dataframe中有2个数据集,我想在同一散点图上可视化它们,所以我尝试了:

import matplotlib.pyplot as plt
import seaborn as sns

sns.pairplot(x_vars=['Std'], y_vars=['ATR'], data=set1, hue='Asset Subclass')
sns.pairplot(x_vars=['Std'], y_vars=['ATR'], data=set2, hue='Asset Subclass')
plt.show()

但是,我一直都得到2张独立的图表,而不是一张 enter image description here 如何可视化同一图上的两个数据集?另外,我对两个数据集都可以使用相同的图例,对第二个数据集使用不同的颜色吗?

1 个答案:

答案 0 :(得分:4)

以下内容应在seaborn(0.9.0)的最新版本中起作用

import matplotlib.pyplot as plt
import seaborn as sns

首先,我们将两个数据集连接为一个,并分配一个dataset列,这将使我们可以保留有关哪个行来自哪个数据集的信息。

concatenated = pd.concat([set1.assign(dataset='set1'), set2.assign(dataset='set2')])

然后,我们使用最新的seaborn版本(0.9.0)中的sns.scatterplot函数,并通过style关键字参数对其进行设置,以使标记基于dataset列:

sns.scatterplot(x='Std', y='ATR', data=concatenated,
                hue='Asset Subclass', style='dataset')
plt.show()