Pandas DataFrame中的编号组

Question

在Pandas中，是否有一种方法可以基于列值对DataFrame中的组进行编号？ 如果我的框架看起来像这样

  Column1 Column2  Column3
0       A       X       23
1       A       X       45
2       A       Y       32
3       A       Y       53
4       A       Y       67
5       B       X       85
6       B       Y       12
7       B       Y       94

我想做的是

df.group_numbers(['Column1', 'Column2'])

  Column1 Column2  Column3  GroupNumber
0       A       X       23            1
1       A       X       45            1
2       A       Y       32            2
3       A       Y       53            2
4       A       Y       67            2
5       B       X       85            3    
6       B       Y       12            4
7       B       Y       94            4

Answer 1

正如ajcr的评论中所建议的ajcr ， pd.factorize是必经之路。 在您的情况下，您可以添加两列以通过添加两列之间带有一些定界符来快速创建键数组。 分隔符是为了避免混淆DSM建议的对，例如ab, c和a, bc 。

df['GroupNumber'] = pd.factorize(df.Column1 + ' ' + df.Column2) 

它仍然比使用pd.lib.fast_zip更快。