首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
社区首页 >问答首页 >寻找两个量之间差异最小的行对

寻找两个量之间差异最小的行对
EN

Stack Overflow用户
提问于 2018-07-12 17:19:17
回答 1查看 375关注 0票数 0

我有一个包含列a和t的数据帧df,其中列"a“具有字符串,列"t”具有整数。我想从dataframe中选择所有行对,对于这些行对,行对的列"a“中的值是相同的,而列"t”中的值的差值具有最小值。例如:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
df = a    t
    abc   4
    abc   3
    def   2
    abc   1

我希望得到以下结果:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
df = a    t
     abc  4
     abc  3

我知道我们可以在同一个数据帧中使用两个for循环,但我正在寻找更有效的解决方案。期待中的感谢

EN

回答 1

Stack Overflow用户

发布于 2018-07-12 17:37:59

您可以使用:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
df = df.sort_values(['a', 't'], ascending=False)
diff_ = df['t']-df['t'].shift(-1)
min_idx = diff_[df['a'] == df['a'].shift(-1)].idxmin()
df.loc[min_idx:min_idx+1]

输出:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
    a   t
0   abc 4
1   abc 3
票数 3
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/51311274

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
查看详情【社区公告】 技术创作特训营有奖征文