首页 > 解决方案 > 通过字符串输入对 DataFrame 列进行数学运算的简单方法?

问题描述

我正在构建一个根据用户的特定需求重新格式化数据的工具。我让用户将规格输入config.json文件。

我正在使用的数据类型之一是 .tsv 文件,为此我使用 python 的 Pandas 模块根据需要读取和操作数据。我希望用户能够使用旧列的数学操作来创建新列。例如:

用户希望使用以下公式将 A 列和 B 列生成 C:

C = (A - B) * 30

所以在config.json他们放:

{"C": "(A - B) * 30"}

我希望将其转换为 python 代码:

df["C"] = (df["A"] - df["B"]) * 30

是否有一个 Pandas 工具可以对列操作进行方程解释?

标签: pythonpandas

解决方案


推荐阅读