python - 如何使用 P 中行值的总和创建一个新列

Question

我有一个有趣的项目，但我被数据清理的一个关键部分困住了:我不知道如何根据特定行/列的值创建总计列

我的数据框有这个信息:

CustomerID ActivityType
01134A       Email
01134B       Email
01134A       Call
01134B       Email

我想要的输出是

 CustomerID ActivityType TotalEmail TotalCall
01134A       Email         1           1
01134B       Email         2           0
01134A       Call          1           1
01134B       Email         2           0

我试过了

df['EmailTotal']='df.groupby('ActivityType'=='Email').transform('sum')

Answer 1

使用DataFrame.join与 crosstab并更改列名称添加 DataFrame.add_prefix :

df = df.join(pd.crosstab(df.CustomerID, df.ActivityType).add_prefix('Total'), on='CustomerID')
print (df)
  CustomerID ActivityType  TotalCall  TotalEmail
0     01134A        Email          1           1
1     01134B        Email          0           2
2     01134A         Call          1           1
3     01134B        Email          0           2

编辑:

您的解决方案应该通过比较、转换为整数和按系列 df.CustomerID 与 transform 和 sum 进行分组来更改:

df['ETotal']=df.ActivityType.eq('Email').astype(int).groupby(df.CustomerID).transform('sum')
df['CTotal']= f.ActivityType.eq('Call').astype(int).groupby(df.CustomerID).transform('sum')