df.query(expr, inplace=False, **kwargs)
是 Pandas 中 DataFrame
对象的一个方法,用于根据指定的布尔表达式查询数据帧的列。
该方法的参数如下:
expr
: 一个包含布尔表达式的字符串或表达式,用于筛选数据帧中的行。inplace
: 一个布尔值,指定是否在原始数据帧上进行修改。如果为True
,则修改原始数据帧;如果为False
(默认值),则返回一个新的数据帧。**kwargs
: 其他可选参数,例如axis
、levels
、sort
等。
下面是一个示例,演示如何使用 df.query()
方法查询数据帧中的特定列:
import pandas as pd
# 创建一个包含姓名和年龄的数据帧
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
'Age': [25, 30, 35, 40]}
df = pd.DataFrame(data)
# 使用 query() 方法查询年龄大于等于 30 的行,并只保留姓名和年龄两列
result = df.query('Age >= 30')[['Name', 'Age']]
print(result)
输出结果为:
Name Age
1 Bob 30
2 Charlie 35
3 David 40
在这个示例中,我们首先创建了一个包含姓名和年龄的数据帧。然后,我们使用 df.query()
方法查询年龄大于等于 30 的行,并使用切片操作 []
只保留了姓名和年龄两列。最后,我们将结果打印出来。