如何高效地按ID分组将DataFrame中相邻行的SEQ列两两组合

轻雪小哥_1764

轻雪小哥_1764

2026-10-16

585人浏览

原创

如何高效地按ID分组将DataFrame中相邻行的SEQ列两两组合

本文介绍一种更简洁高效的Pandas方法,利用groupby().shift()实现同ID下SEQ值的成对拼接(如"1 - 2"、"2 - 3"),避免冗余列和显式过滤,显著提升可读性与执行效率。

本文介绍一种更简洁高效的pandas方法,利用`groupby().shift()`实现同id下seq值的成对拼接(如"1 - 2"、"2 - 3"),避免冗余列和显式过滤,显著提升可读性与执行效率。

在处理有序分组序列数据时,常需将同一分组内相邻记录的某列值进行配对组合(例如生成区间、路径边、状态转移对等)。原始方案通过shift(-1)创建下一行副本、添加临时标识列、再用布尔索引过滤,逻辑清晰但存在三处可优化点:① 引入冗余列(ID_NEXT、SEQ_NEXT)增加内存开销;② x['ID']==x['ID_NEXT'] 过滤依赖隐式对齐,易受索引扰动影响;③ 类型转换与字符串拼接分散在多步中,链式可读性弱。

推荐使用分组内前向位移 + 链式赋值的一体化方案:

import pandas as pd

x = pd.DataFrame({
    'ID':  [1, 1, 1, 1, 2, 2, 2],
    'SEQ': [1, 2, 3, 4, 1, 3, 4]
})

result = (x.astype({'SEQ': 'str'})
          .assign(
              SEQ_NEXT=lambda df: df.groupby('ID')['SEQ'].shift(),  # 同ID内上一行SEQ(即前驱)
              COMBINE=lambda df: df['SEQ_NEXT'] + ' - ' + df['SEQ']
          )
          .dropna(subset=['SEQ_NEXT'])  # 自动剔除每组首行(无前驱)
          [['ID', 'SEQ', 'SEQ_NEXT', 'COMBINE']])  # 可选:精简输出列

print(result)

输出:

   ID SEQ SEQ_NEXT COMBINE
1   1   2        1   1 - 2
2   1   3        2   2 - 3
3   1   4        3   3 - 4
5   2   3        1   1 - 3
6   2   4        3   3 - 4

✅ 关键优势解析:

  • groupby('ID')['SEQ'].shift() 确保位移严格限定在相同ID组内,天然规避跨组错误配对(如ID=1末尾与ID=2开头的误连);
  • dropna(subset=['SEQ_NEXT']) 直接丢弃每组首行(其SEQ_NEXT为NaN),比布尔索引更语义明确且性能更优;
  • 全程链式操作(.assign().dropna())无需中间变量,内存友好,符合函数式编程习惯。

⚠️ 注意事项:

  • 若原始SEQ为数值型且需保留计算能力,建议仅对拼接步骤临时转字符串(如 df['SEQ'].astype(str)),而非全局修改原列类型;
  • 此方法默认生成“前驱→当前”对(即 SEQ[i-1] - SEQ[i]),若需“当前→后继”对(SEQ[i] - SEQ[i+1]),应改用 shift(-1) 并配合 dropna,但需注意此时末行会被丢弃;
  • 数据必须已按 ID 和 SEQ(或业务顺序)预排序,否则分组位移结果无意义。

该方案以更少代码、更强健的分组逻辑和更清晰的数据流,成为处理同类问题的标准实践。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

2025.12.04

320

15

Python 数据清洗与预处理实战
Python 数据清洗与预处理实战

本专题系统讲解 Python 在数据清洗与预处理中的核心技术,包括使用 Pandas 进行缺失值处理、异常值检测、数据格式化、特征工程与数据转换,结合 NumPy 高效处理大规模数据。通过实战案例,帮助学习者掌握 如何处理混乱、不完整数据,为后续数据分析与机器学习模型训练打下坚实基础。

2026.01.31

252

21

Python数据分析与Pandas高级实战
Python数据分析与Pandas高级实战

本专题围绕 Python 数据分析展开,系统讲解 Pandas 的高级用法,包括数据清洗、透视表、时间序列分析以及多表合并与分组操作。通过实战案例,帮助开发者掌握高效处理与分析数据的方法,提高数据处理效率与分析能力。

2026.04.13

511

25

Revel框架模板渲染与表单验证
Revel框架模板渲染与表单验证

本专题围绕Revel框架视图层开发,介绍模板语法、变量输出、布局复用、表单提交、参数绑定与验证规则,帮助开发者完成安全规范的页面交互。

2026.10.16

0

15

Revel框架MVC路由与控制器
Revel框架MVC路由与控制器

本专题聚焦Revel框架MVC核心机制,讲解路由定义、控制器Action、参数绑定、请求响应和反向路由,并结合常见场景梳理接口请求处理流程。

2026.10.16

0

15

Revel框架入门与安装配置
Revel框架入门与安装配置

本专题从Revel框架环境搭建讲起,系统介绍命令行工具安装、项目创建、目录结构、配置文件及热代码重载,帮助开发者快速完成首个Go Web应用。

2026.10.16

0

15

Linux 新手先学哪些命令
Linux 新手先学哪些命令

Linux新手命令专题汇集终端基础、文件目录操作、文本查看、软件安装和权限管理等入门文章,帮助按实际任务练习常用命令并建立命令行操作习惯。

2026.10.15

0

20

MySQL 忘记密码怎么办
MySQL 忘记密码怎么办

MySQL忘记密码专题汇集不同版本与系统下的root密码重置方法,覆盖MySQL 5.7、8.0、Mac、Linux及本地集成环境,并提示恢复权限与重新登录的检查步骤。

2026.10.15

0

18

Vue.js 中 ref 和 reactive 的区别
Vue.js 中 ref 和 reactive 的区别

Vue.js中ref和reactive的区别专题汇集两种响应式API的定义、访问方式、对象解构、侦听与性能取舍等文章,帮助结合实际状态类型选择合适写法。

2026.10.15

0

20

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.7万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.7万人学习