gpt4 book ai didi

python - pandas groupby,然后按列的值选择一行(例如,最小值、最大值)

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

假设我有一个数据框

Category Data1 column1
A 'SOMEDATA' 10
A 'SOMEDATA' 2
A 'SOMEDATA' -10
B 'SOMEDATA' 10
B 'SOMEDATA' 1
B 'SOMEDATA' -10

等等

我想按列值在每个组中选择一行。例如,ABS(column1)

所以结果数据是
Category Data1 column1
A 'SOMEDATA' 2
B 'SOMEDATA' 1

我怎么能在python中做到这一点?

我不知道如何返回整行。例如,
df.groupby('Category')['column1'].min();

这只会返回 'Category' min(column1) 。

最佳答案

这是一个计算效率更高的解决方案。

TL;DR 版本

df.loc[df.groupby('Category')['column1'].idxmin()]

关于python - pandas groupby,然后按列的值选择一行(例如,最小值、最大值),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55186262/

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