0

0

如何高效地按行计算 Pandas Series 中的动态表达式(基于变量字典)

霞舞

霞舞

发布时间:2026-01-21 20:11:16

|

137人浏览过

|

来源于php中文网

原创

如何高效地按行计算 Pandas Series 中的动态表达式(基于变量字典)

本文介绍一种高性能方法,将含变量名的表达式 series(如 `"a + b"`)与对应变量值字典结合,通过前向填充和安全表达式求值,逐行计算出结果 series。适用于大规模(200+ series)场景,兼顾可读性与执行效率。

金融建模、时间序列推演或规则引擎等场景中,常需根据动态定义的公式(如 "A + B - C")对齐时间索引并逐行计算结果。核心挑战在于:公式稀疏不连续、变量值以独立 Series 存储、需自动继承上一行有效公式、且整体性能敏感

以下为推荐实现方案,关键步骤包括:

  1. 公式前向填充(ffill):确保每行都拥有待执行的完整表达式;
  2. 构建统一 DataFrame:将公式列与所有变量 Series 横向拼接,对齐索引;
  3. 安全表达式重写与求值:用正则将 A → r['A'],使 eval() 可直接访问当前行数据;
  4. 逐行应用 apply():避免 Python 循环,利用 Pandas 底层优化。

✅ 完整示例代码:

import pandas as pd
import re

# 输入:公式 Series(索引为年份)
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 & 2:公式前向填充 + 合并变量数据
df = pd.concat([
    eqns.reset_index(name='Equation').ffill(),  # 生成带 index 和 Equation 列的 DataFrame
    pd.DataFrame(values)                        # 自动对齐索引(缺失值补 NaN)
], axis=1)

# 步骤 3 & 4:安全重写表达式并逐行求值
df['Result'] = df.apply(
    lambda r: eval(re.sub(r'([A-Za-z_]\w*)', r"r['\1']", r['Equation'])), 
    axis=1
)

# 提取最终结果 Series(保留原始索引,仅 Result 列)
result_series = df.set_index('index')['Result']
print(result_series)

? 输出结果:

拍我AI
拍我AI

AI视频生成平台PixVerse的国内版本

下载
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 或 simpleeval);
  • 性能优化:对 200+ Series 批量处理时,建议将全部公式与变量一次性构建成大 DataFrame,避免重复 apply;
  • 索引对齐:务必确保 values 中各 Series 的索引与 eqns.index 覆盖范围一致(推荐显式指定 index=),否则 pd.concat 可能引入意外 NaN;
  • 数字类型兼容性:eval() 返回 Python 原生类型,Pandas 会自动转为 float64;如需保持整数精度,可在最后调用 .astype("Int64")(支持 NaN 的整数类型)。

该方案在千级数据规模下实测耗时

相关专题

更多
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相关的文章、下载、课程内容,供大家免费下载体验。

709

2023.08.11

Java编译相关教程合集
Java编译相关教程合集

本专题整合了Java编译相关教程,阅读专题下面的文章了解更多详细内容。

9

2026.01.21

热门下载

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

精品课程

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

共4课时 | 10.7万人学习

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号