python - 在 Pandas 中分组和转换数据

Question

我是 python 的新手，我想使用 pandas 转换数据框。

原始数据框是:

A1,B1,C1,M
A1,B1,C1,HM
A2,B2,C2,M
A2,B2,C2,M
A3,B3,C3,

目标是:

A1,B1,C1,HM
A2,B2,C2,M
A3,B3,C3,

我需要删除重复项并根据值 A、B、C 计算最后一个字段 (HM/M)。我想知道如何才能实现这种转变。

Answer 1

IIUC，您可以sort_values 并只选择每个group 的first()，因为M 排在HM 之后

df.sort_values(['A','B','C','H']).groupby(['A','B','C']).first() 

    A   B   C   H
0   A1  B1  C1  HM
1   A2  B2  C2  M
2   A3  B3  C3  NaN