如何在同一图中绘制由分类字段选择的两个分布?

 收藏

我有一个带有一个分类字段和几个连续值字段的数据框。 我希望绘制具有两个曲线的这些连续字段之一的分布,每个分类值一个。 我可以绘制两个这样的单一分布,但是无法在同一图中绘制它们两个。

以下是演示数据和当前结果的代码:

import pandas as pd
import numpy as np
samples = 200
# class can be 0 or 1; value is the continuous value
df = pd.DataFrame({'value' : np.random.randn(samples),
                   'class' : np.random.randint(0, high=2, size=samples)})
df.head()

df['value'][df['class'] == 1].plot.kde()
df['value'][df['class'] == 0].plot.kde()

上面的代码有效,但是绘制了两条单独的曲线。我需要在同一张图片中。 我尝试使用concat,但是由于该系列具有不同的索引,因此会加载NaN。 谢谢。

回复