python - 基于另一个字段创建计数器

Question

我想创建下表。

所需表格

ID     Coverage   Count
1      A          1
1      A          2
1      A          3
1      B          1
2      C          1
2      A          1
2      A          2
2      C          2

我目前表中只有 ID 和 coverage，需要添加 count 列。我希望它从 1 开始，对同一 ID 下的相同覆盖范围计数 +1。

在 SQL 中，它将涉及按 ID 和 Coverage 进行分组，但我不确定在 python 中从哪里开始。

Answer 1

这里需要cumcount

df['Newcount']=df.groupby(['ID','Coverage']).cumcount()+1
df
Out[588]: 
   ID Coverage  Count  Newcount
0   1        A      1         1
1   1        A      2         2
2   1        A      3         3
3   1        B      1         1
4   2        C      1         1
5   2        A      1         1
6   2        A      2         2
7   2        C      2         2