数据筛选背景
在处理数据时,我们可能希望从数据库里面筛选出符合特定条件的记录(个案或样本,不同的行业叫法不一样),平常大家对Excel筛选很熟悉,比如从A字段(变量或特征)包含“团队”,B字段大于等于40,筛选出符合这两个条件的记录,如下图所示:
excel数据筛选
pandas处理
正确代码
#加载库
import pandas as pd
import numpy as np
#读取数据
data=pd.read_excel('test.xlsx')
#查看数据类型
data.dtypes
#查看数据前5行
data.head()
#根据条件筛选出数据
data.loc[(data['A'].str.contains('团队')) & (data['B']>=40)]
#data[(data['A'].str.contains('团队')) & (data['B']>=40)] #这两行都可以
结果示例
错误代码
1.位运算符&与|,而不是逻辑运算符 and 与or,两者是有区别的
data.loc[(data['A'].str.contains('团队')) and (data['B']>=