python使用query函数报错：UndefinedVariableError

导入excel表数据

import pandas as pd
df1 = pd.read_excel('面板数据.xlsx',sheet_name='城市统计年鉴')
df1.head()

问题描述

df1.query('城市==临沂市')

UndefinedVariableError: name '临沂市' is not defined

• 多次尝试后发现是因为python将“临沂市”识别为变量了，解决方法是将其变为字符串。

解决办法

df1.query('城市=="临沂市"') # 选择多个城市可以使用：df1.query('城市=="临沂市" or 城市=="七台河市"')

选取特定数值的数据时，比如选取2008年数据，使用：df1.query(‘年份==2008’)即可。