python-3.x - 计算另一列中 Id 的总行数

Question

我有一个数据框

初始化列表数据。

data = {'Id':['1', '2', '3', '4','5','6','7','8','9','10'], 'reply_id':[2, 2,2, 5,5,6,8,8,1,1]} 

创建数据框

df = pd.DataFrame(data)

   Id   reply_id
0   1   2
1   2   2
2   3   2
3   4   5
4   5   5
5   6   6
6   7   8
7   8   8
8   9   1
9   10  1

我想在 new 中为每个 Id 获取总的 reply_id。

Id=1 在 reply_id 中出现了 2 次，我希望在新列 new

中出现

期望的输出

    Id  reply_id  new
0   1   2          2   
1   2   2          3     
2   3   2          0 
3   4   5          0
4   5   5          2
5   6   6          1
6   7   8          0
7   8   8          2
8   9   1          0
9   10  1          0

我已经完成了这行代码。

df['new'] = df.reply_id.eq(df.Id).astype(int).groupby(df.Id).transform('sum')

Answer 1

在这个答案中，我使用 Series.value_counts 来计算 reply_id 中的值，并将结果转换为字典。然后，我在 Id 列上使用 Series.map 将计数与 Id 相关联。 fillna(0) 用于填充 reply_id

中不存在的值

df['new'] = (df['Id']
             .astype(int)
             .map(df['reply_id'].value_counts().to_dict())
             .fillna(0)
             .astype(int))