如何在 Pandas 中高效地为每 N 行生成滑动列表并赋值给新列

雨敏姑娘_7089

雨敏姑娘_7089

2026-09-24

185人浏览

原创

如何在 Pandas 中高效地为每 N 行生成滑动列表并赋值给新列

本文介绍使用 numpy.sliding_window_view 替代低效列表推导式或受限的 rolling().apply(),实现将 DataFrame 某列的滑动窗口数据(如每 3 行)转为列表并存入新列,兼顾性能与可读性。

本文介绍使用 `numpy.sliding_window_view` 替代低效列表推导式或受限的 `rolling().apply()`,实现将 dataframe 某列的滑动窗口数据(如每 3 行)转为列表并存入新列,兼顾性能与可读性。

在 Pandas 中,当需要基于滑动窗口(如每 3 行)将某列的值聚合为 Python 列表并写入新列时,常见的误区是尝试用 rolling(window=3).apply(lambda x: list(x)) —— 这会报错 TypeError: must be real number, not list。原因在于 rolling().apply() 默认要求返回标量数值(用于统计计算),不支持直接返回非标量对象(如 list),即使设置 raw=False 也无法绕过该限制。

更高效、更规范的解决方案是借助 NumPy 1.20+ 引入的 sliding_window_view,它专为生成滑动窗口视图而设计,返回一个形状为 (n - window + 1, window) 的二维数组,再通过 .tolist() 可直接转为嵌套列表,最后封装为 pd.Series 并对齐索引即可。

以下为完整实现示例:

import pandas as pd
import numpy as np

# 构造示例数据
np.random.seed(150)
df = pd.DataFrame(np.random.randint(0, 10, size=(10, 2)), columns=['A', 'B'])

# 使用 sliding_window_view 生成长度为 3 的滑动列表列 'C'
from numpy.lib.stride_tricks import sliding_window_view as swv

N = 3
windowed_lists = swv(df['B'], N).tolist()  # → [[9,2,5], [2,5,9], ..., [1,9,7]]
df['C'] = pd.Series(windowed_lists, index=df.index[N-1:])  # 自动对齐:前 N-1 行为 NaN

print(df)

输出结果中,前两行(索引 0 和 1)的 'C' 值为 NaN(因窗口不足),从索引 2 开始填充对应三元列表,与预期完全一致。

⚠️ 注意事项:

  • sliding_window_view 要求 NumPy ≥ 1.20;旧版本可升级 pip install --upgrade numpy;
  • 返回的 Series 默认以 df.index[N-1:] 为索引,未覆盖部分自动补 NaN,无需手动填充;
  • 若需保留 None(而非 NaN),可在赋值后执行:df['C'] = df['C'].where(df['C'].notna(), None);
  • 相比原始列表推导式(O(n²) 时间复杂度),此方法为 O(n),在百万级数据上性能提升显著。

总结:当 apply 遇到非标量返回限制时,应跳出 Pandas 单一 API 思维,善用 NumPy 底层向量化工具。sliding_window_view 不仅语义清晰、代码简洁,更是处理滑动列表类任务的推荐实践。

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

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

下载

相关标签:

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

相关专题

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

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

2025.12.04

300

15

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

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

2026.01.31

212

21

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

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

2026.04.13

311

25

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

2026.09.23

60

15

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

2026.09.23

20

15

Buffalo框架零基础入门教程
Buffalo框架零基础入门教程

本专题整理Buffalo框架入门内容,涵盖Go环境准备、buffalo CLI安装、新项目生成、目录结构说明、dev热加载启动、数据库连接配置与常见报错排查,帮助新手按约定优于配置的思路跑通第一个Buffalo框架应用。

2026.09.23

20

15

Conan创建软件包配方指南
Conan创建软件包配方指南

本专题介绍通过conanfile.py创建软件包的方法,讲解包名、版本、依赖和构建设置等基础信息,以及source、build、package、package_info等常用方法的作用及编写思路。

2026.09.22

20

12

Conan二进制包配置指南
Conan二进制包配置指南

本专题介绍Conan根据操作系统、编译器、架构和构建类型生成二进制包的方法,讲解Profile、Settings、Options及Package ID的作用,帮助管理不同平台和编译环境下的包版本。

2026.09.22

20

13

Conan私有仓库搭建教程
Conan私有仓库搭建教程

本专题系统的讲解Conan私有仓库的搭建流程,涵盖仓库服务部署、存储目录配置、用户认证、权限划分和远程地址添加,并介绍内部C++依赖包的上传、下载及版本维护方法。

2026.09.22

20

19

热门下载

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

精品课程

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

共6课时 | 54.6万人学习

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

共89课时 | 133.2万人学习