我有以下df，它是从here改编而来的
import pandas as pd
df = pd.DataFrame({'group':[1,1,1,2,2,2,3,3,3],
'value':[1,2,3,1,2,3,4,3,2]})
我希望得到以下结果:
df1 = pd.DataFrame({'group':[1,1,1,2,2,2,3,3,3],
'value':[0,2,3,0,2,3,4,3,0]})
逻辑是(1) for each 组 Select 前2个数字，(2)将其他值设置为0.
有什么建议吗？谢谢
使用nlargest/组进行布尔索引:
m = df.groupby('group')['value'].apply(lambda g: g.isin(g.nlargest(2)))
df['value'] = df['value'].where(m, 0)
输出(为清晰起见，作为新列"value2")
group value value2
0 1 1 0
1 1 2 2
2 1 3 3
3 2 1 0
4 2 2 2
5 2 3 3
6 3 4 4
7 3 3 3
8 3 2 0