python中query的用法

在 Python 的 Pandas 库中，可以使用 `query()` 方法来从 DataFrame 中筛选出符合特定条件的行。`query()` 方法接受一个字符串参数，该参数包含一个表达式，用于指定要筛选的条件。

以下是一些示例，演示如何使用 `query()` 方法：

1. 筛选列值满足条件的行

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({'name': ['Alice', 'Bob', 'Charlie'], 'age': [25, 30, 35]})

# 使用 query() 方法筛选出 age 列值大于 30 的行
result = df.query('age > 30')
print(result)

输出结果为：

       name  age
2  Charlie   35

2. 筛选多个条件的行

# 使用 query() 方法筛选出 age 列值在 25 到 35 之间，且 name 列值为 'Bob' 的行
result = df.query('age >= 25 and age <= 35 and name == "Bob"')
print(result)

输出结果为：

   name  age
1   Bob   30

3. 使用变量作为条件

# 定义一个变量 age_threshold，并使用它作为条件筛选行
age_threshold = 30
result = df.query('age > @age_threshold')
print(result)

输出结果为：

       name  age
2  Charlie   35

需要注意的是，在查询表达式中，可以使用 `@` 符号来引用变量，如上面的示例所示。此外，查询表达式还可以使用其他常用的运算符和函数，如 `==`、`!=`、`<`、`>`、`in`、`not in`、`startswith()`、`endswith()` 等。