按ID分组，并按每个ID计算唯一开始日期。转换回dataframe。熊猫

Question

我有一个熊猫数据框架：

pd.DataFrame({
    'student_id': ['5', '5', '5', '5', '2', '2'],
    'start_date': ['2020-11-11', '2020-11-11', '2020-11-11', '2020-12-05', '2020-11-25', '2020-11-25']
})

我想按'student_id‘对df进行分组，并计算每行出现多少个相同的开始日期。然后，我想将其转换回原始数据帧。例如，我将创建一个名为'course_enroll_count‘的新列。前三行将显示3，因为学生有3个开始日期'2020-11-11‘。预期输出：

pd.DataFrame({
    'student_id': ['5', '5', '5', '5', '2', '2'],
    'start_date': ['2020-11-11', '2020-11-11', '2020-11-11', '2020-12-05', '2020-11-25', '2020-11-25'],
    'course_enroll_count': [3, 3, 3, 1, 2, 2]
})

Answer 1

尝试使用transform

df['new'] = df.groupby(['student_id','start_date'])['start_date'].transform('count')
df
Out[313]: 
  student_id  start_date  new
0          5  2020-11-11    3
1          5  2020-11-11    3
2          5  2020-11-11    3
3          5  2020-12-05    1
4          2  2020-11-25    2
5          2  2020-11-25    2