如何用一行代码连接两个数据框

Question

我正在解决 Titanic Kaggle 代码并使用 pandas 进行数据分析。数据在这里 ( https://www.kaggle.com/competitions/titanic/data )

我有两个 Dataframe，我想用聚合代码连接它们。

train_df[['Parch', 'Survived']].groupby(['Parch'], as_index=False).count().rename(columns={'Survived':"Count"}).reset_index(drop=True)

train_df[['Parch', 'Survived']].groupby(['Parch'], as_index=False).mean().rename(columns={'Survived':"Surviving rate"}).reset_index(drop=True)

而我将这些代码与列方向联系起来的结论就是这样。 但是我觉得下面的代码很晕。 还有另一种方法可以用一行代码简单地连接这两个数据帧。

tmp_df = train_df[['Parch','Survived']].groupby(['Parch'], as_index=False).agg({"Survived": ["count","mean"]}).reset_index(drop=True)

tmp_df.columns = tmp_df.columns.droplevel()
tmp_df.columns = ["Parch", "Count", "Surviving rate"]

如何用一行简单的代码将这两个数据框与列方向连接起来？

Answer 1

使用命名聚合可以更简洁地编写您的语句：

tmp_df = (
    train_df.groupby("Parch")
    .agg(**{"Count": ("Survived", "count"), "Surviving rate": ("Survived", "mean")})
    .reset_index()
)