列名中带有空格的python数据框查询

Question

我想使用 query 过滤数据框

ExcludeData= [1,3,4,5]
dfResult.query('Column A in @ExcludeData')

如何在查询中使用 A 列而不重命名它？

Answer 1

从 Pandas v. 0.25 开始，如果在查询中将列名括在反引号中，则可以引用名称中包含空格的列。

使用 Pandas 0.25.2:

>>> df = pd.DataFrame({'a a': [1, 0], 'b b': [1, 1]})
>>> df
   a a  b b
0    1    1
1    0    1
>>> df.query('`a a`==`b b`')
   a a  b b
0    1    1

来自 API 文档:https://pandas.pydata.org/pandas-docs/version/0.25/reference/api/pandas.DataFrame.query.html

在你的情况下，用法是:

dfResult.query('`Column A` in @ExcludeData')