当前位置:首页 > 行业动态 > 正文

pandas where 多条件

在pandas中,我们可以使用where()函数来根据多个条件过滤数据。where()函数的基本语法如下:

DataFrame.where(condition, other=None)

参数说明:

condition:一个布尔数组,用于指定哪些行应该被保留。

other:可选参数,如果指定了其他值,那么在条件为False的位置将使用这些值,默认值为None。

下面是一个使用where()函数的示例:

假设我们有一个名为df的DataFrame,包含以下数据:

姓名 年龄 性别
张三 25
李四 30
王五 28
赵六 22

现在我们想要筛选出年龄大于等于25岁且性别为男的记录,可以使用以下代码:

import pandas as pd
data = {'姓名': ['张三', '李四', '王五', '赵六'],
        '年龄': [25, 30, 28, 22],
        '性别': ['男', '女', '男', '女']}
df = pd.DataFrame(data)
使用where()函数筛选年龄大于等于25岁且性别为男的记录
result = df.where((df['年龄'] >= 25) & (df['性别'] == '男'))
print(result)

输出结果:

姓名 年龄 性别
张三 25
王五 28
0

随机文章