189 8069 5689

Python中怎么利用pandas实现按条件选择

Python中怎么利用pandas实现按条件选择,相信很多没有经验的人对此束手无策,为此本文总结了问题出现的原因和解决方法,通过这篇文章希望你能解决这个问题。

创新互联坚持“要么做到,要么别承诺”的工作理念,服务领域包括:网站制作、网站建设、企业官网、英文网站、手机端网站、网站推广等服务,满足客户于互联网时代的高邑网站设计、移动媒体设计的需求,帮助企业找到有效的互联网解决方案。努力成为您成熟可靠的网络建设合作伙伴!


numpy.where 方法

Excel 函数中有一个初学者都能马上学会的函数——IF 函数,而在 pandas 中却没有对应效果的方法,这是因为 numpy 已经有了对应的实现—— where。

他能根据条件(true 或者 false) 返回不同的值。由于需要使用 numpy 的方法,因此代码的开始需要导入 numpy 包:

1import pandas as pd
2import numpy as np
 
 

场景

如下学生成绩表:

Python中怎么利用pandas实现按条件选择

  • 高于等于60分算合格,C列打上"是",否则打上"否"

  • 典型的根据条件选择某个值的需求

 

怎么解决

如此简单的需求,Excel 中一个 IF 函数轻松解决:

Python中怎么利用pandas实现按条件选择

  • IF 函数第一参数是条件,第二参数是当第一条件为 true 时的返回,第三参数是当第一条件为 false 时的返回

在使用 numpy.where 方法时的逻辑与上述 Excel 的 IF 函数一致:

Python中怎么利用pandas实现按条件选择

1df = pd.read_excel('data.xlsx', 'sp1')
2df['res'] = np.where(df.成绩>=60,'是','否')
3df
 
  • 行2:np.where 各个参数都能接受 pandas 的列(Series)


 

性能优越

如果你看过本系列文章会发现,怎么当初入门 Python 的时候,学习的各种处理列表、字典的技巧全都用不上了。甚至 if、for 循环都很少用到了。

在 pandas 中其实也可以选择用 Python 的基本语法处理。

比如上面的例子,我们可以用 apply:

Python中怎么利用pandas实现按条件选择

1df = pd.read_excel('data.xlsx', 'sp1')
2
3def ap_where(x):
4    if x >= 60:
5        return '是'
6    return '否'
7
8df['res'] = df.成绩.apply(ap_where)
9df
 

看完上述内容,你们掌握Python中怎么利用pandas实现按条件选择的方法了吗?如果还想学到更多技能或想了解更多相关内容,欢迎关注创新互联行业资讯频道,感谢各位的阅读!


新闻标题:Python中怎么利用pandas实现按条件选择
文章起源:http://cdxtjz.com/article/ijjdcs.html

其他资讯