使用 pandas.DataFrame.query抽取数据

# 抽出a列==1的行
df.query('a == 1') 

# 抽出a列!=1的行
df.query('a != 1') 

# 使用字符串指定条件时，值必须用双引号 (") 括起来
df.query('a == "hello" ') 

# 使用多个条件和逻辑运算符（例如 and 和 or）进行条件提取
df.query('a > 2 and b < 3') 
df.query('a > 2 or b < 3')
df.query('a > 2 or e == "d" ') 

# 有点between and 的意思
df.query(' 2<= a <= 4')

# 如果该值有缺失值，则按如下方式提取NaN的行
df.query(' d != d') 
# 抽取非NaN的行
df.query(' d == d')  

# 如果该值具有布尔值，例如 True 或 False，则按如下方式提取
df.query('f == True') 
df.query('f == False') 

# IN 和 NOT IN
df.query(' a in (1, 2, 3)') 
df.query(' a not in (1, 2, 3)') 

# 使用变量
x = 2
df.query(' a == @x')

y = [1, 2, 3] 
df.query('a in @y') 

z = "hello"
df.query(' a == @z')