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

Question

在泰坦尼克号数据集中，我希望计算每个乘客类别（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'))

所以，我得到以下输出：

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

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

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

提前致谢！

标签： pythonpandasaggregation

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