python - 使用条件 groupby 计算分类列的百分比并在 Python 中计数

Question

我想计算每个 ID 的所有行中真值的百分比。

这是我的数据示例:

id     col1    
 1     True
 1     True
 1     False
 1     True
 2     False
 2     False

新列应如下所示:

id     col1    num_true
 1     True     0.75
 1     True     0.75
 1     False    0.75
 1     True     0.75
 2     False    0
 2     False    0

这是我尝试做的:

df['num_true']= df[df['col1'] == 'True'].groupby('id')['col1'].count()
df['num_col1_id']= df.groupby('id')['col1'].transform('count')

df['perc_true']= df.num_true/df.num_col1_id

Answer 1

groupby 并应用transform 得到mean

df['num_true']=df.groupby('id').col1.transform('mean')



  id   col1  num_true
0   1   True      0.75
1   1   True      0.75
2   1  False      0.75
3   1   True      0.75
4   2  False      0.00
5   2  False      0.00