gpt4 book ai didi

python - 使用 GroupBy 在 DataFrame 中查找最小值

转载 作者:行者123 更新时间:2023-11-28 21:37:44 25 4
gpt4 key购买 nike

我有这样的数据:

Index SourceID  ComparedID  Distance
571 106 8 0.004846
572 106 11 0.000000
573 106 23 0.012712
574 107 84 0.002800
575 107 70 0.007555

它位于 pandas 数据框中。我想弄清楚如何做到这一点:

我想从找到最小距离的数据帧创建一个结果并返回 ComparedID

所以输出看起来像:

SourceID ComparedID Distance
106 11 0.000000
107 84 0.002800

我该怎么做?

最佳答案

你可以使用 idxmin()

df.iloc[df.groupby('SourceID')['Distance'].idxmin(), :]

Index SourceID ComparedID Distance
1 572 106 11 0.0000
3 574 107 84 0.0028

关于python - 使用 GroupBy 在 DataFrame 中查找最小值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49073656/

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