pandas if语句
时间: 2024-03-21 15:36:13 浏览: 53
if语句使用
在pandas中,没有专门的if语句来进行条件判断。但是可以使用条件表达式和布尔索引来实现类似的功能。
下面是一个示例,演示如何使用条件表达式和布尔索引来进行条件判断:
```python
import pandas as pd
# 创建一个示例DataFrame
data = {'Name': ['Tom', 'Nick', 'John', 'Amy'],
'Age': [20, 25, 30, 35]}
df = pd.DataFrame(data)
# 使用条件表达式进行条件判断
df['IsAdult'] = df['Age'].apply(lambda x: 'Yes' if x >= 18 else 'No')
# 使用布尔索引筛选满足条件的行
adults = df[df['IsAdult'] == 'Yes']
print(df)
print(adults)
```
输出结果:
```
Name Age IsAdult
Tom 20 Yes
1 Nick 25 Yes
2 John 30 Yes
3 Amy 35 Yes
Name Age IsAdult
0 Tom 20 Yes
1 Nick 25 Yes
2 John 30 Yes
3 Amy 35 Yes
```
在上面的示例中,我们使用了条件表达式`'Yes' if x >= 18 else 'No'`来判断年龄是否大于等于18岁,并将结果存储在新的列`IsAdult`中。然后,我们使用布尔索引`df['IsAdult'] == 'Yes'`来筛选出满足条件的行,即年龄大于等于18岁的行。
阅读全文