在 python 数据框中使用 groupby 的 where 语句

Question

train[['Pclass', 'Age']].groupby(['Pclass'], as_index=False).median().sort_values(by='Pclass', ascending=True)

这是我进行分组、查找汇总统计数据并根据列（在本例中为“Pclass”）对其进行排序的地方。

如何同时使用 where 子句？我想输入的 where 子句将执行类似于 train[train.Survived==1]

对如何实现这一点有任何想法吗？我正在使用经典的“泰坦尼克号”数据集。

Answer 1

将train[['Pclass', 'Age']]改为

train.loc[train['Survived'] == 1, ['Pclass', 'Age']]

---

例如，

import pandas as pd
import seaborn as sns
train = sns.load_dataset("titanic")

print(train.loc[train['survived'] == 1, ['pclass', 'age']]
           .groupby(['pclass'], as_index=False)
           .median()
           .sort_values(by='pclass', ascending=True))

打印

   pclass   age
0       1  35.0
1       2  28.0
2       3  22.0