一尘不染

将局部变量与pandas eval函数配合使用

python

熊猫帮助文件说(用于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未定义局部变量“ mult”


阅读 263

收藏
2021-01-20

共1个答案

一尘不染

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

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

但是,这确实具有创建额外列的副作用。

更新:

我已经提交了具有此问题的修复程序的请求请求

2021-01-20