Pandas的查询函数在列名中存在空格时无法正常工作。

12 浏览
0 Comments

Pandas的查询函数在列名中存在空格时无法正常工作。

我有一个包含列名中带有空格的数据帧。我试图使用query方法来获取结果。对于'c'列，它工作正常，但对于'a b'列会出现错误。

import pandas as pd
a = pd.DataFrame(columns=["a b", "c"])
a["a b"] = [1,2,3,4]
a["c"] = [5,6,7,8]
a.query('a b==5')

对于此问题，我收到以下错误信息：

a b ==5
  ^
SyntaxError: invalid syntax

我不想用其他字符（如'_'等）填充空格。

有一个使用pandasql的技巧，将变量名放在方括号内，例如：[a b]。

0