gpt4 book ai didi

python - 迭代 Pandas groupby 对象的子集

转载 作者:行者123 更新时间:2023-12-01 00:59:00 25 4
gpt4 key购买 nike

我有一个 Pandas groupby 对象,我想迭代前 n 个组。我试过:

import pandas as pd
df = pd.DataFrame({'A':['a','a','a','b','b','c','c','c','c','d','d'],
'B':[1,2,3,4,5,6,7,8,9,10,11]})

df_grouped = df.groupby('A')
i = 0
n = 2 # for instance
for name, group in df_grouped:
#DO SOMETHING
if i == n:
break
i += 1

group_list = list(df_grouped.groups.keys())[:n]
for name in group_list:
group = df_grouped.get_group(name)
#DO SOMETHING

但我想知道是否有更优雅/Pythonic 的方法来做到这一点?

我的实际 groupby 中有 1000 个组,我只想对一个子集执行操作,只是为了获得整个数据的印象。

最佳答案

您可以使用原始 df 进行过滤,然后我们可以完成您需要做的所有其他操作

yourdf=df[df.groupby('A').ngroup()<=1]
<小时/>
yourdf=df[pd.factorize(df.A)[0]<=1]

关于python - 迭代 Pandas groupby 对象的子集,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55971179/

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