熊猫查询功能不适用于列名中的空格

Question

我有一个列名中有空格的数据框。我正在尝试使用 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

我不想用“_”等其他字符填充空间。

有一个 hack 使用 pandasql 将变量名放在括号内示例： [ab]

原文由 Bhushan Pant 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

从 pandas 0.25 开始，您将能够使用反引号转义列名，这样您就可以做到

a.query('`a b` == 5')

原文由 Jarno 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

恐怕查询方法不接受带有空格的列名。在任何情况下，您都可以通过以下方式查询数据框：

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

原文由 DTT 发布，翻译遵循 CC BY-SA 4.0 许可协议