python - 根据值的第 n 次重复创建 Pandas 列

Question

我有一个像这样的数据框:

userId  dateTime

2233    2018-08-01 12:30:29
1455    2018-08-01 13:08:01
2233    2018-08-01 15:08:01
1909    2018-08-01 15:18:01
2233    2018-08-01 17:58:01
1455    2018-08-01 19:08:01

所以我想根据同一用户出现的第 n 次添加一个新列。所以我想要的数据框将如下所示:

userId  dateTime             Appearance_no

2233    2018-08-01 12:30:29    1
1455    2018-08-01 13:08:01    1
2233    2018-08-01 15:08:01    2
1909    2018-08-01 15:18:01    1
2233    2018-08-01 17:58:01    3
1455    2018-08-01 19:08:01    2

这几乎与序列号类似，但我无法生成该列。

Answer 1

您可以使用 cumcount

df['Appearance_no'] = df.groupby('userId')['userId'].cumcount() + 1

    userId  dateTime              Appearance_no
0   2233    2018-08-01 12:30:29   1
1   1455    2018-08-01 13:08:01   1
2   2233    2018-08-01 15:08:01   2
3   1909    2018-08-01 15:18:01   1
4   2233    2018-08-01 17:58:01   3
5   1455    2018-08-01 19:08:01   2