是否可以在 python 中的 df[ ] 中放置一个字符串？

Question

我知道这很糟糕。我真的很想知道这是否可以在 python 中完成，所以我有两个带有算术方程的字符串，现在我想将它们放在 df[] 中。 df是一个数据框这可以吗？

X = "'cars'+'bikes'*'planes'"

现在应该像下面这样放置

X = df['cars']+df['bikes']*df['planes']

如果可能怎么做？

Answer 1

我假设您知道使用 eval 的后果。

s =  "'cars'+'bikes'*'planes'"

df['out'] = eval(re.sub(r"([^+\-*\/]+)", r'df[\1]', s))

它所做的基本上是替代df。它将'cars'+'bikes'*'planes' 更改为df['cars']+df['bikes']*df['planes']。 如果不想使用eval，可以解析column names 和operands 之类的

columns = re.findall(r"'([^+\-*\/]+)'", s)
operands = re.findall(r'([+\-*\/]+)', s)

但在这种情况下，您需要定义操作优先级并创建一个树来计算结果。

---

更新

import re
import pandas as pd

s =  "'cars'+30*'bikes'-'planes'+20"
s2 = re.sub(r"('[^+\-*\/'\d]+')", r'df[\1]', s)

pd.eval(s2)