我有以下数据集:
id window Rank member
1 2 2 0
1 3 2 0
2 3 1 0
2 2 1 0
我想让成员在window==3时等于Rank。为此,我有以下命令:
df["member"]= df[df['window']==3]['Rank'][0]
但是,我想在按 id 分组的 groupby 语句中执行此操作。以下命令返回错误。这可能是我在这里缺少的一件简单的事情,但我无法解决如何在上面的命令中使用 groupby 的问题。非常感谢任何帮助。
df["member"]= df.groupby("id")[df[df['window']==3]['Rank'][0]]
最佳答案
您可以通过使用pandas.DataFrame.where
来实现这一点 -
df = pd.DataFrame({'id':[1,1,2,2],'window':[2,3,3,2],'Rank':[2,2,1,1],'member':[0,0,0,0]})
=>
Rank id member window
0 2 1 0 2
1 2 1 0 3
2 1 2 0 3
3 1 2 0 2
df['member'] = df['Rank'].where(df['window']==3, df['member'])
print(df)
=>
Rank id member window
0 2 1 0 2
1 2 1 2 3
2 1 2 1 3
3 1 2 0 2
关于python - Pandas:Groupby 中的语法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44257750/