pandas DataFrame 根据多列的值做判断,生成新的列值实例
更新时间：2018年05月18日 14:37:00 作者：魏之燕
今天小编就为大家分享一篇pandas DataFrame 根据多列的值做判断,生成新的列值实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
环境：Python3.6.4 + pandas 0.22
主要是DataFrame.apply函数的应用，如果设置axis参数为1则每次函数每次会取出DataFrame的一行来做处理，如果axis为1则每次取一列。
如代码所示，判断如果城市名中含有ing字段且年份为2016，则新列test值赋为1，否则为0.
import numpy as np
import pandas as pd
data = {'city': ['Beijing', 'Shanghai', 'Guangzhou', 'Shenzhen', 'Hangzhou', 'Chongqing'],
'year': [2016,2016,2015,2017,2016, 2016],
'population': [2100, 2300, 1000, 700, 500, 500]}
frame = pd.DataFrame(data, columns = ['year', 'city', 'population', 'debt'])
def function(a, b):
if 'ing' in a and b == 2016:
return 1
else:
return 0
print(frame, '\n')
frame['test'] = frame.apply(lambda x: function(x.city, x.year), axis = 1)
print(frame)
运行结果如下：
另外Series类型也有apply函数，用法示例如下：
import numpy as np
import pandas as pd
data = {'city': ['Beijing', 'Shanghai', 'Guangzhou', 'Shenzhen', 'Hangzhou', 'Chongqing'],
'year': [2016,2016,2015,2017,2016, 2016],
'population': [2100, 2300, 1000, 700, 500, 500]}
frame = pd.DataFrame(data, columns = ['year', 'city', 'population', 'debt'])
print(frame, '\n')
frame['panduan'] = frame.city.apply(lambda x: 1 if 'ing' in x else 0)
print(frame)
运行结果如下：
以上这篇pandas DataFrame 根据多列的值做判断,生成新的列值实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章给大家介绍了如何利用ctypes提高Python的执行速度，对大家学习使用python具有一定的参考借鉴价值。有需要的朋友们一起来看看吧。2016-09-09
这篇文章主要介绍了python兼容VBA的用法详解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下2021-09-09
今天小编就为大家分享一篇Python利用heapq实现一个优先级队列的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-02-02
动态图形是使可视化更具吸引力和用户吸引力的好方法,它帮助我们以有意义的方式展示数据可视化,本文将利用Matplotlib实现绘制一些常用动态图形,希望对大家有所帮助2024-02-02
七夕快到了,所以本文小编将给给大家介绍五种拿来就能用的炫酷表白代码,无限弹窗表白,爱心发射,心动表白,玫瑰花等表白代码,需要的小伙伴快来试试吧2023-08-08
这篇文章主要介绍了Django项目如何给数据库添加约束，帮助大家更好的理解和学习使用Django框架，感兴趣的朋友可以了解下2021-04-04
pandas是基于NumPy的一种工具，该工具是为了解决数据分析任务而创建的。Pandas纳入了大量库和一些标准的数据模型，提供了高效操作大型数据集的工具。pandas提供大量快速便捷地处理数据的函数和方法。你很快就会发现，它是使Python强大而高效的数据分析环境的重要因素之一2021-11-11
这篇文章主要介绍了Python编程实现粒子群算法(PSO)详解，涉及粒子群算法的原理，过程，以及实现代码示例，具有一定参考价值，需要的朋友可以了解下。2017-11-11
Python使用jsonpath-rw模块处理Json对象操作示例
这篇文章主要介绍了Python使用jsonpath-rw模块处理Json对象操作,结合实例形式分析了Python使用requests与response处理json的方法,并给出了jsonpath_rw模块操作json对象的基本示例,需要的朋友可以参考下2018-07-07
pdfminer是一个开源的Python第三方库,专门用于解析PDF文件,本文主要为大家详细介绍了如何使用pdfminer实现PDF文本提取,有需要的小伙伴可以了解下2025-02-02
最新评论