根据输入在pandas中循环实现多次条件查询,并给出用例
时间: 2024-02-25 17:52:06 浏览: 61
多条件查询示例
假设我们有一个包含学生信息的 Pandas DataFrame,包括学生姓名、性别、年龄、班级和成绩。现在我们想要根据不同的条件查询学生信息,可以使用 Pandas 中的 loc 函数来实现。以下是一个简单的用例:
```python
import pandas as pd
# 创建一个包含学生信息的 DataFrame
data = {'Name': ['Tom', 'Jerry', 'Lucy', 'Lily', 'John', 'David'],
'Gender': ['M', 'M', 'F', 'F', 'M', 'M'],
'Age': [18, 19, 17, 18, 18, 19],
'Class': ['A', 'B', 'A', 'B', 'A', 'B'],
'Score': [80, 75, 85, 90, 70, 95]}
df = pd.DataFrame(data)
# 多次条件查询
# 查询班级为 A,成绩大于等于80分的男生
male_A = df.loc[(df['Class'] == 'A') & (df['Gender'] == 'M') & (df['Score'] >= 80)]
# 查询年龄为 18 岁,班级为 B 的学生
age_18_B = df.loc[(df['Age'] == 18) & (df['Class'] == 'B')]
# 查询成绩排名前三的学生
top_3 = df.sort_values(by='Score', ascending=False).iloc[:3]
```
上述代码中,我们先创建了一个包含学生信息的 DataFrame。然后,我们使用 loc 函数进行多次条件查询。第一次查询是查询班级为 A,成绩大于等于80分的男生;第二次查询是查询年龄为 18 岁,班级为 B 的学生;第三次查询是查询成绩排名前三的学生。最后,我们将查询结果存储在不同的变量中,方便后续使用。
阅读全文