python - 有条件地替换 pandas DataFrame 中行的值

Question

这是我的 Pandas 数据框:

ID  START   END  SEQ
1   11      12   5
1   14      15   6 
1   13      14   7 
2   10      14   1
3   11      15   1
3   16      17   2

我想更改 SEQ 的值，以便对于相同的 ID，SEQ 值将是 1、2...等，例如

ID  START   END  SEQ
1   11      12   1
1   14      15   3 
1   13      14   2 
2   10      14   1
3   11      15   1
3   16      17   2

我怎样才能有效地避免 for 循环？

Answer 1

在 groupby 操作上使用 cumcount:

df.groupby('ID').cumcount()+1