文章/答案/技术大牛

发布

社区首页 >问答首页 >pandas.core.groupby.DataFrameGroupBy.idxmin()非常慢，我怎样才能让我的代码更快呢？

问pandas.core.groupby.DataFrameGroupBy.idxmin()非常慢，我怎样才能让我的代码更快呢？
EN

Stack Overflow用户

提问于 2019-05-01 14:47:01

回答 1查看 330关注 0票数 1

我正在尝试执行与SQL group by相同的操作，并取最小值：

select id,min(value) ,other_fields...
from table
group by ('id')

我试过了：

dfg = df.groupby('id', sort=False)
idx = dfg['value'].idxmin()
df = df.loc[idx, list(df.columns.values)]

但是第2行https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.core.groupby.DataFrameGroupBy.idxmin.html ()在df中的~4M列上花费了半个多小时，而group by只用了不到1秒，我错过了什么，它应该花那么长时间吗？怎样才能让这个过程更快呢？在纯SQL中会更快吗？

回答已采纳

发布于 2019-05-01 14:51:24

df1 = df.sort_values(by=['value']).drop_duplicates('id', keep='first')

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/55932560

复制

相似问题

问pandas.core.groupby.DataFrameGroupBy.idxmin()非常慢，我怎样才能让我的代码更快呢？EN