gpt4 book ai didi

python - Pandas Groupby 只取前 N 组

转载 作者:太空狗 更新时间:2023-10-29 18:02:17 31 4
gpt4 key购买 nike

<分区>

我有一些 DataFrame,我想按 ID 分组,e。例如:

import pandas as pd
df = pd.DataFrame({'item_id': ['a', 'a', 'b', 'b', 'b', 'c', 'd'], 'user_id': [1,2,1,1,3,1,5]})
print df

生成:

  item_id  user_id
0 a 1
1 a 2
2 b 1
3 b 1
4 b 3
5 c 1
6 d 5

[7 rows x 2 columns]

我可以轻松地按 id 分组:

grouped = df.groupby("item_id")

但是我怎样才能只返回前 N 个分组对象呢?例如我只想要前 3 个唯一的 item_id。

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com