Pandas:Groupby 中的语法

弗雷迪888

我有以下数据集:

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

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章