使用Pandas在同一图中绘制分组数据

Question

在熊猫,我正在做:

bp = p_df.groupby('class').plot(kind='kde')

p_df是一个dataframe对象.

然而,这产生了两个图,每个类一个.如何在同一个图中强制同时使用两个类的一个图？

Answer 1

版本1:

您可以创建轴,然后使用ax关键字DataFrameGroupBy.plotto将所有内容添加到这些轴:

import matplotlib.pyplot as plt

p_df = pd.DataFrame({"class": [1,1,2,2,1], "a": [2,3,2,3,2]})
fig, ax = plt.subplots(figsize=(8,6))
bp = p_df.groupby('class').plot(kind='kde', ax=ax)

这是结果:

不幸的是,传说的标签在这里没有太大意义.

版本2:

另一种方法是循环遍历组并手动绘制曲线:

classes = ["class 1"] * 5 + ["class 2"] * 5
vals = [1,3,5,1,3] + [2,6,7,5,2]
p_df = pd.DataFrame({"class": classes, "vals": vals})

fig, ax = plt.subplots(figsize=(8,6))
for label, df in p_df.groupby('class'):
    df.vals.plot(kind="kde", ax=ax, label=label)
plt.legend()

这样您就可以轻松控制图例.这是结果:

Answer 2

另一种方法是使用seaborn模块.这将绘制相同轴上的两个密度估计,而不指定用于保持轴的变量,如下所示(使用另一个答案中的一些数据框设置):

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
%matplotlib inline

# data to create an example data frame
classes = ["c1"] * 5 + ["c2"] * 5
vals = [1,3,5,1,3] + [2,6,7,5,2]
# the data frame 
df = pd.DataFrame({"cls": classes, "indices":idx, "vals": vals})

# this is to plot the kde
sns.kdeplot(df.vals[df.cls == "c1"],label='c1');
sns.kdeplot(df.vals[df.cls == "c2"],label='c2');

# beautifying the labels
plt.xlabel('value')
plt.ylabel('density')
plt.show()

这导致以下图像.

Answer 3

import matplotlib.pyplot as plt
p_df.groupby('class').plot(kind='kde', ax=plt.gca())

Answer 4

• 有两种简单的方法可以在同一图中绘制每个组。
  1. 使用 时pandas.DataFrame.groupby，应指定要绘制的列（例如聚合列）。
  2. 使用seaborn.kdeplotorseaborn.displot并指定hue参数
• 使用pandas v1.2.4，matplotlib 3.4.2，seaborn 0.11.1
• OP 特定于绘制，但对于许多绘图类型（例如、 等），kde步骤是相同的​​。kind='line'sns.lineplot

导入和样本数据

• 对于样本数据，组位于'kind'列中，并且将绘制 的 ，kde忽略。'duration''waiting'

import pandas as pd
import seaborn as sns

df = sns.load_dataset('geyser')

# display(df.head())
   duration  waiting   kind
0     3.600       79   long
1     1.800       54  short
2     3.333       74   long
3     2.283       62  short
4     4.533       85   long

绘图与pandas.DataFrame.plot

• .groupby使用或重塑数据.pivot

.groupby

• 指定聚合列 、['duration']和kind='kde'。

ax = df.groupby('kind')['duration'].plot(kind='kde', legend=True)

.pivot

ax = df.pivot(columns='kind', values='duration').plot(kind='kde')

绘图与seaborn.kdeplot

• 指定hue='kind'

ax = sns.kdeplot(data=df, x='duration', hue='kind')

绘图与seaborn.displot

• 指定hue='kind'和kind='kde'

fig = sns.displot(data=df, kind='kde', x='duration', hue='kind')

阴谋

Answer 5

也许你可以尝试这个：

fig, ax = plt.subplots(figsize=(10,8))
classes = list(df.class.unique())
for c in classes:
    df2 = data.loc[data['class'] == c]
    df2.vals.plot(kind="kde", ax=ax, label=c)
plt.legend()