使用带有pandas eval函数的多个赋值的局部变量

时间:2017-02-06 23:33:21

标签: python python-3.x pandas

pandas帮助文件说(对于eval): 为方便起见,可以使用多行字符串执行多项分配。

但是,我发现它不适用于变量(使用ipython):

这有效:

df_price.eval("op = op * @mult", inplace = True)

但这不起作用(op,cl,hi,lo是数据帧df_price中的cols,其中mult是浮点数):

df_price.eval("""op = op * @mult
              cl = cl * @mult
              hi = hi * @mult
              lo = lo * @mult""", inplace = True)

错误: pandas.computation.ops.UndefinedVariableError:local variable' mult'未定义

1 个答案:

答案 0 :(得分:1)

我可以确认局部变量似乎只适用于多行eval表达式的第一行。可能的解决方法:

df_price.eval("""mult = @mult
              op = op * mult
              cl = cl * mult
              hi = hi * mult
              lo = lo * mult""", inplace = True)

但这会产生额外列的副作用。

<强>更新

我已提交Pull Request修正此问题。