使用查询.DataFrame.query具有列名中具有特殊字符的数据帧

Question

我注意到，从版本0.25开始，在使用pandas.DataFrame.query时，列名中允许有空格，也就是说，这些列名应该用反勾号括起来。例如：

import pandas as pd

df = pd.DataFrame({'a b':[1,0,1,1,0,0],
                   'c d':[1,0,1,1,0,0],
                   'e f':[0,0,0,0,1,0]})

print(df)

   a b  c d  e f
0    1    1    0
1    0    0    0
2    1    1    0
3    1    1    0
4    0    0    1
5    0    0    0

q = "(`a b` == 1) | (`c d` == 1) | (`e f` == 1)"
df = df.query(q)

print (df)

   a b  c d  e f
0    1    1    0
2    1    1    0
3    1    1    0
4    0    0    1

它工作正常，但我的列可能包含符号，加号或其他特殊字符。他们目前似乎没有得到支持：

df2 = pd.DataFrame({'a b+':[1,0,1,1,0,0],
                   'c | d':[1,0,1,1,0,0],
                   'e & f':[0,0,0,0,1,0]})

print(df2)

   a b+  c | d  e & f
0     1      1      0
1     0      0      0
2     1      1      0
3     1      1      0
4     0      0      1
5     0      0      0

q = "(`a b+` == 1) | (`c | d` == 1) | (`e & f` == 1)"
df2 = df2.query(q)

print (df2)

上次打印给我一个错误：

Traceback (most recent call last):
  File "C:\Users\XXX\AppData\Local\Programs\Python\Python37\lib\site-packages\pandas\core\computation\scope.py", line 188, in resolve
    return self.resolvers[key]
  File "C:\Users\XXX\AppData\Local\Programs\Python\Python37\lib\collections\__init__.py", line 914, in __getitem__
    return self.__missing__(key)            # support subclasses that define __missing__
  File "C:\Users\XXX\AppData\Local\Programs\Python\Python37\lib\collections\__init__.py", line 906, in __missing__
    raise KeyError(key)
KeyError: 'a_b_'

是否有任何解决方法或其他方法来为数据帧构建过滤条件？我想定义一个将动态过滤器返回为字符串的函数。你知道吗

Answer 1

实际上，熊猫支持列名中的unicode符号，它应该可以工作。 请尝试此方法：

set1 = df2['a b+'] == 1
set2 df2['c | d'] == 1

print(df2[set1 | set2])

在你的测试数据上为我工作