当列名以数字开头时，Pandas 查询会抛出错误

我正在尝试对以下数据框执行查询：

data = {'ab': [1,2,3], 'c1': [1,2,3], 'd': [1,2,3], 'e_f': [1,2,3]}
df = pd.DataFrame(data)
for cl in df.columns:
    print len(df.query('%s==2' %cl))

这很好用。但是，如果列名以数字开头，则会引发语法错误。

data = {'ab': [1,2,3], 'c1': [1,2,3], '1d': [1,2,3], 'e_f': [1,2,3]}
df = pd.DataFrame(data)
for cl in df.columns:
    print len(df.query('%s==2' %cl))

文件“”，第 1 行

1 d ==2
  ^

语法错误：语法无效

我认为问题与字符串的格式有关。我想知道形成此查询的正确方法是什么。

query uses pandas.eval，即有记录的 http://pandas.pydata.org/pandas-docs/dev/generated/pandas.eval.html#pandas.eval“将 Python 表达式作为字符串求值”。您的查询不是有效的 Python 表达式，因为1d在Python中不是有效的语法，所以你不能使用query以这种方式引用本专栏。

如果你确保所有列都是有效的 Python 标识符，pandas 中的事情通常会更容易。