在Pandas中,query函数用于从数据集中查询满足特定条件的行。它接受一个字符串参数,该参数描述了要查询的条件。
以下是使用query函数的基本语法:
df.query('condition')
其中,df是要查询的数据集,condition是查询条件。
例如,假设我们有一个名为df的数据集,其中包含以下列:name,age,gender和salary。要查询年龄大于30岁的所有行,可以使用以下代码:
result = df.query('age > 30')
这将返回一个新的数据集result,其中包含满足条件的所有行。
除了使用比较运算符(如大于号、小于号等),还可以使用其他逻辑运算符(如AND、OR、NOT等)来组合多个条件。例如,要查询年龄大于30岁且性别为女性的所有行,可以使用以下代码:
result = df.query('age > 30 & gender == "female"')
这将返回一个新的数据集result,其中包含满足条件的所有行。