pandas实现对一列/多列进行数据区间筛选

更新时间：2024年02月22日 08:57:53 作者：icestone_kai

这篇文章主要介绍了pandas实现对一列/多列进行数据区间筛选方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

下面这个表

如果你想筛选出age大于等于18，小于等于30的行：

如果在mysql中整这种活很容易，用个where就可以了

在pandas中也有相当于where作用的语法，loc

筛选18<=age<=30的数据

import pandas as pd

stu = pd.read_csv("data/student.csv", index_col='id')

# 留下18<=年龄<=30
def age_18_to30(a):
    return 18 <= a <= 30

# 留下 85<=score
def level_a(s):
    return 85 <= s

# 使用loc会生成一个新的series
stu = stu.loc[stu['age'].apply(age_18_to30)]
# 或者用下lambda表达式:
# stu = stu.loc[stu['age'].apply(lambda a:18<=a<=30)]

print(stu)

结果：

此时再加一个筛选条件

age大于等于18，小于等于30且分数大于等于85：

代码：

import pandas as pd

stu = pd.read_csv("data/student.csv", index_col='id')


# 留下18<=年龄<=30
def age_18_to30(a):
    return 18 <= a <= 30


# 留下 85<=score
def level_a(s):
    return 85 <= s

stu = stu.loc[stu['age'].apply(age_18_to30)].loc[stu['score'].apply(level_a)]

print(stu)

结果：

其中，获取莫一列，我们一直使用的是stu['age']，这个还可以写为：stu.age

然后就整活成功了！

文件：

F:\Project\python\src\WangYiYun\DataAnalysis\19_.py

完整代码：

# @DATE : 2021-1-2
# @TIME : 16:13
# @USER : kirin
import pandas as pd

stu = pd.read_csv("data/student.csv", index_col='id')

# 留下18<=年龄<=30
def age_18_to30(a):
    return 18 <= a <= 30

# 留下 85<=score
def level_a(s):
    return 85 <= s

# 使用loc会生成一个新的series
# stu = stu.loc[stu['age'].apply(age_18_to30)]
stu = stu.loc[stu['age'].apply(age_18_to30)].loc[stu['score'].apply(level_a)]
# 或者不使用 stu['age'] ：
# stu = stu.loc[stu.age.apply(age_18_to30)].loc[stu.score.apply(level_a)]

# 使用lambda表达式:
# stu = stu.loc[stu.age.apply(lambda a: 18 <= a <= 30)].loc[stu.score.apply(lambda s: 85 <= s)]

# 代码太长回个车:(空格+反斜线+回车)
# stu = stu.loc[stu.age.apply(lambda a: 18 <= a <= 30)]. \
#     loc[stu.score.apply(lambda s: 85 <= s)]

print(stu)

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

深入理解Python装饰器
装饰器(decorator)是一种高级Python语法。装饰器可以对一个函数、方法或者类进行加工。这篇文章主要介绍了深入理解Python装饰器的相关资料,需要的朋友可以参考下
2016-07-07
python学习实操案例(三)
这篇文章主要介绍了python学习实操案例，循环输出26个字母对应的ASCII码值、模拟用户登录、猜数游戏练习等案例，需要的小伙伴可以参考一下，希望对你有一定的帮助
2022-02-02
Python数据分析numpy文本数据读取索引切片实例详解
这篇文章主要为大家介绍了Python数据分析numpy文本数据读取索引切片实例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2023-08-08
python实现大战外星人小游戏实例代码
这篇文章主要介绍了python实现大战外星人小游戏,本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下
2019-12-12
Python+radar实现随机日期时间的生成
Python有广泛丰富的第三方库，在没有特殊定制下，避免了重复造轮子。本文将利用radar库实现生成随机的日期或时间，文中的示例代码讲解详细，感兴趣的可以了解一下
2022-05-05
python实现批量视频分帧、保存视频帧
这篇文章主要为大家详细介绍了python实现批量视频分帧、保存视频帧，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-05-05
python3中编码获取网页的实例方法
在本篇文章里小编给大家整理了一篇关于python3中编码获取网页的实例方法，有兴趣的朋友们可以学习下。
2020-11-11
Python趣味实例，实现一个简单的抽奖刮刮卡
这篇文章主要介绍了Python如何实现一个简单的抽奖刮刮卡，文中讲解非常细致，代码帮助大家更好的理解和学习，感兴趣的朋友可以了解下
2020-07-07
Python3爬虫中识别图形验证码的实例讲解
在本篇内容里小编给大家分享的是关于Python3爬虫中识别图形验证码的实例讲解内容，需要的朋友们可以学习参考下。
2020-07-07
关于torch.scatter与torch_scatter库的使用整理
这篇文章主要介绍了关于torch.scatter与torch_scatter库的使用整理,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2023-09-09