问对列名中包含特殊字符的数据帧使用pandas.DataFrame.query

EN

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_'