如何在 Pandas 中使用 groupby 划分 2 个聚合列？

Question

在 Titanic 数据集中，我希望计算每个乘客 class（Pclass）1,2 和 3 中幸存的乘客百分比。我想出了如何计算乘客数量和数量。 使用 group by 幸存的乘客如下：

train[['PassengerId','Pclass','Survived']]\
    .groupby('Pclass')\
    .agg(PassengerCount=pd.NamedAgg(column='PassengerId', aggfunc='count'), 
         SurvivedPassengerCount=pd.NamedAgg(column='Survived',aggfunc='sum'))

所以，我得到以下 output：

        PassengerCount  SurvivedPassengerCount
Pclass                                        
1                  216                     136
2                  184                      87
3                  491                     119

但是我如何获得百分比列？ 我想要 output 如下：

        PassengerCount  SurvivedPassengerCount  PercSurvived
Pclass                                                             
1                  216                     136                     62.9%
2                  184                      87                     47.3%
3                  491                     119                     24.2%

提前致谢！

Answer 1

由于您只需要将SurvivedPassengerCount除以PassengerCount ，因此可以使用.assign方法执行此操作：

result = train[['PassengerId','Pclass','Survived']]\
            .groupby('Pclass')\
            .agg(PassengerCount=pd.NamedAgg(column='PassengerId', aggfunc='count'), 
                 SurvivedPassengerCount=pd.NamedAgg(column='Survived',aggfunc='sum'))\
    
result = result.assign(PercSurvived=df['PassengerCount']/df['SurvivedPassengerCount'])