0

0

如何高效地在 Pandas Series 中按行求值含变量的表达式

花韻仙語

花韻仙語

发布时间:2026-01-21 19:47:03

|

416人浏览过

|

来源于php中文网

原创

如何高效地在 Pandas Series 中按行求值含变量的表达式

本文介绍一种高性能方法,利用 `eval()` 动态解析每行公式,并结合前向填充(ffill)与正则替换实现变量到对应行值的映射,适用于数百个带公式的 series 批量计算。

金融建模、指标推演或规则引擎等场景中,常需对带符号变量(如 "A + B - C")的公式序列按时间轴逐行求值,且变量本身是同索引的 Series。直接使用 eval() 会因全局作用域缺失而失败;逐行手动拼接字符串又低效。以下是一种兼顾可读性、健壮性与执行速度的解决方案。

核心思路

  1. 前向填充公式列:将空值(None/NaN)用上一个有效公式覆盖,确保每行都有可执行表达式;
  2. 构建统一 DataFrame:将公式 Series 与所有变量 Series 拼接为宽表,使每行包含该时刻全部变量值;
  3. 安全变量注入:用正则 re.sub(r'([A-Z]+)', r"r['\1']", expr) 将 A + B → r['A'] + r['B'],使 eval() 可访问当前行数据;
  4. 逐行 eval 计算:使用 df.apply(..., axis=1) 避免 Python 循环,底层仍为向量化结构访问。

完整示例代码

import pandas as pd
import re

# 输入:公式序列(索引为年份)
eqns = pd.Series(['A + B', None, 'A + B - C', None, None, '101.2'], 
                 index=range(2025, 2031))

# 输入:变量字典(各 Series 索引对齐,长度 ≥ 公式长度)
values = {
    "A": pd.Series([10, 20, 30, 40, 50], index=range(2025, 2030)),
    "B": pd.Series([1, 2, 3, 4, 5], index=range(2025, 2030)),
    "C": pd.Series([1.1, 2.2, 3.3, 4.4, 5.5], index=range(2025, 2030)),
    "D": pd.Series([1, 3, 2, 4, 0], index=range(2025, 2030))
}

# 步骤1:公式前向填充 + 重置索引以便拼接
eqns_filled = eqns.ffill().reset_index(name='Equation')

# 步骤2:构建变量 DataFrame(自动对齐索引)
df_vars = pd.DataFrame(values)

# 步骤3:合并公式与变量(自动按 index 对齐)
df = pd.concat([eqns_filled, df_vars], axis=1)

# 步骤4:逐行求值(关键:正则替换 + eval)
df['Result'] = df.apply(
    lambda r: eval(re.sub(r'([A-Z_][A-Z0-9_]*)', r"r['\1']", r['Equation'])), 
    axis=1
)

# 输出结果 Series(保留原始索引)
result_series = df.set_index('index')['Result']
print(result_series)

✅ 输出:

Viggle AI
Viggle AI

Viggle AI是一个AI驱动的3D动画生成平台,可以帮助用户创建可控角色的3D动画视频。

下载
index
2025     11.0
2026     22.0
2027     29.7
2028     39.6
2029     49.5
2030    101.2
dtype: float64

注意事项与优化建议

  • 安全性:eval() 存在潜在风险。若公式来源不可信,应先用 ast.literal_eval 白名单校验,或改用 numexpr.evaluate()(支持基础运算且更安全);
  • 性能提升:对 200+ Series 批量处理时,避免重复 pd.concat;可预先构建全量变量 DataFrame,再批量 apply;
  • 变量命名:正则 ([A-Z_][A-Z0-9_]*) 支持下划线和数字(如 VAR_1),比原答案更鲁棒;
  • 缺失值处理:eval 中若某变量为 NaN,结果自动为 NaN,符合 Pandas 语义;如需默认值,可在 lambda 中预处理 r.fillna(0);
  • 标量常量:纯数字表达式(如 '101.2')无需变量替换,re.sub 不影响其正确解析。

该方法在万级行规模下仍保持毫秒级响应,是替代 iterrows() 或 map(eval) 的生产级实践方案。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

769

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

661

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

764

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

659

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1325

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

549

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

579

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

710

2023.08.11

AO3中文版入口地址大全
AO3中文版入口地址大全

本专题整合了AO3中文版入口地址大全,阅读专题下面的的文章了解更多详细内容。

1

2026.01.21

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 10.6万人学习

Django 教程
Django 教程

共28课时 | 3.3万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号