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']]

---

例如

^{pr2}$

印刷品

   pclass   age
0       1  35.0
1       2  28.0
2       3  22.0