我有一个如下的DataFrame：
print(df)
Product Color Weight
0 A Red 13.01
1 A Red 13.04
2 A Red 13.10
3 A Red 13.11
我想删除重复项并仅保存重量为 max() 的产品。
print(df)
Product Color Weight
0 A Red 13.11
谢谢
【问题讨论】：
标签： python-3.x pandas duplicates
您可以将groupby 与.max 一起使用
#if you don't care about color remove it from the groupby clause.
#df.groupby(['Product'])['Weight'].max().reset_index()
df1 = df.groupby(['Product','Color'])['Weight'].max().reset_index()
print(df1)
Product Color Weight
0 A Red 13.11
【讨论】：
df.sort_values('Weight').drop_duplicates(['Product', 'Color'], keep='last')
drop_duplicates performance = 197 ms ± 1.77 ms per loop (mean ± std. dev. of 7 runs, 1 loop each) 和 groupby performance = 108 ms ± 1.23 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)。