python - 如何使用pandas创建组ID

Question

R 的 data.table 包有一个非常方便的 .GRP 方法来生成组索引值。

library(data.table)
dt <- data.table(
  Grp=c("a", "z", "a", "f", "f"),
  Val=c(3, 2, 1, 2, 2)
)
dt[, GrpIdx := .GRP, by=Grp]

   Grp Val GrpIdx
1:   a   3      1
2:   z   2      2
3:   a   1      1
4:   f   2      3
5:   f   2      3

使用 pandas 完成同样事情的最佳方法是什么？

import pandas as pd
df = pd.DataFrame({'Grp':["a", "z", "a", "f", "f"], 'Val':[3, 2, 1, 2, 2]})

Answer 1

对于 Pandas >= 1.1，您可以使用 groupby.ngroup()。

在您的示例中:

In [39]: df['GrpIdx'] = df.groupby(['Grp']).ngroup()    

In [40]: df                                                                                   
Out[40]: 
  Grp  Val  Grpidx
0   a    3       0
1   z    2       2
2   a    1       0
3   f    2       1
4   f    2       1