python - Pandas - 对所有列进行分组并在原始数据框中进行标记

Question

这个问题已经有答案了:

Pandas: assign an index to each group identified by groupby (6 个回答)

已关闭 4 年前。

我有一个DataFrame带列'Id'这是独一无二的，并且 'A', 'B', 'C'等等...

有不同的行，其中所有值 'A', 'B', 'C'是相同的。我想给他们一个组名(从 1 开始的运行索引)。

例如:

df = pd.DataFrame({"A": [1, 1, 1, 2], "B": [3, 4, 4, 4], "C": [5, 5, 5, 5]})
df
Out[127]: 
   A  B  C
0  1  3  5
1  1  4  5
2  1  4  5
3  2  4  5

将成为

   A  B  C  grp
0  1  3  5    1
1  1  4  5    2
2  1  4  5    2
3  2  4  5    3

我知道我可以分组['A', 'B', 'C']并获取 key ，但是，我必须以未优化的方式迭代 key 和数据帧。我未能以优化的方式做到这一点

Answer 1

使用GroupBy.ngroup :

df['grp'] = df.groupby(['A', 'B', 'C']).ngroup() + 1
print (df)

   A  B  C  grp
0  1  3  5    1
1  1  4  5    2
2  1  4  5    2
3  2  4  5    3

如果列已排序:

df['grp'] = pd.factorize([tuple(x) for x in df.values])[0] + 1