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 的评论中所建议的，pd.factorize 是要走的路。在您的情况下，您可以添加两列以通过添加两列并在它们之间添加一些分隔符来快速创建一个键数组。分隔符是为了避免混淆对，例如ab, c 和a, bc，正如DSM 所建议的那样。

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

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