pandas按条件筛选数据loc
时间: 2023-10-17 18:02:35 浏览: 151
Pandas的loc函数可以用于按照条件筛选数据。loc函数用于基于标签的索引,它可以按照行和列的标签筛选数据。
假设我们有一个名为 df 的数据框,其中包含以下数据:
| | name | age | gender |
|---:|:-------|:------|:---------|
| 0 | Alice | 25 | Female |
| 1 | Bob | 30 | Male |
| 2 | Charlie| 35 | Male |
| 3 | Dave | 40 | Male |
| 4 | Eve | 45 | Female |
我们可以使用以下代码按照条件筛选数据:
```
# 选择性别为 Female 的行
df.loc[df['gender'] == 'Female']
# 选择年龄大于 30 的行
df.loc[df['age'] > 30]
# 选择名字为 Bob 的行
df.loc[df['name'] == 'Bob']
# 选择名字为 Bob 或 Alice 的行
df.loc[(df['name'] == 'Bob') | (df['name'] == 'Alice')]
```
这些代码将返回一个包含符合条件的行的数据框。我们也可以使用 loc 函数选择特定的列:
```
# 选择名字和年龄两列
df.loc[:, ['name', 'age']]
# 选择名字为 Bob 或 Alice,年龄和性别两列
df.loc[(df['name'] == 'Bob') | (df['name'] == 'Alice'), ['age', 'gender']]
```
这些代码将返回一个包含符合条件的列的数据框。
阅读全文