Python：如何查找和替换字符串中的所有变量名

Question

假设我从前端得到一个字符串，例如

str='(A==1) & (B==\'A\') & (C>sin(2))'

这是最简单的格式，字符串可能要复杂得多。

我想在数据帧过滤中应用条件，例如

data = {'A': [1, 2, 3, 4],\
        'B': ['A','B','C','D'],\
        'C':[0.1,0.2,0.3,0.4]}
df=pd.DataFrame(data)
df_test=df[eval(str)]

为了完成这项工作，我必须在字符串中找到变量A,B,C ，并用df.A, df.B, df.C.替换它们。

我试过以下方法

import ast
names = [node.id for node in ast.walk(ast.parse(str)) if isinstance(node, ast.Name)]
print(names)

但它返回['C', 'A', 'B', 'sin']其中 'sin' 不是必需的。

我也尝试过pyparse但仍然无法弄清楚如何定义变量名的模式。

如果您能帮助我就如何查找和替换字符串中的变量名提供一些建议，我们将不胜感激？

Answer 1

您可以使用ast.NodeTransformer进行替换：

import ast
s = '(A==1) & (B==\'A\') & (C>sin(2))'
data = {'A': [1, 2, 3, 4], 'B': ['A', 'B', 'C', 'D'], 'C': [0.1, 0.2, 0.3, 0.4]}
class toDf(ast.NodeTransformer):
    def visit_Name(self, node):
       if node.id in data: #check if variable name exists in the data
           node = ast.Attribute(value=ast.Name(id='df'), attr=node.id)
       return node

new_s = ast.unparse(toDf().visit(ast.parse(s)))
print(new_s)

输出：

(df.A == 1) & (df.B == 'A') & (df.C > sin(2))