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更快。