在pandas中遍历DataFrame行的实现方法


Posted in Python onOctober 23, 2019

有如下 Pandas DataFrame:

import pandas as pd
inp = [{'c1':10, 'c2':100}, {'c1':11,'c2':110}, {'c1':12,'c2':120}]
df = pd.DataFrame(inp)
print df

上面代码输出:

   c1   c2
0  10  100
1  11  110
2  12  120

现在需要遍历上面DataFrame的行。对于每一行,都希望能够通过列名访问对应的元素(单元格中的值)。也就是说,需要类似如下的功能:

for row in df.rows:
 print row['c1'], row['c2']

Pandas 可以这样做吗?

我找到了similar question。但这并不能给我需要的答案,里面提到:

for date, row in df.T.iteritems():

要么

for row in df.iterrows():

但是我不明白row对象是什么,以及我如何使用它。

最佳解决方案

要以 Pandas 的方式迭代遍历DataFrame的行,可以使用:

DataFrame.iterrows()

for index, row in df.iterrows():
 print row["c1"], row["c2"]

DataFrame.itertuples()

for row in df.itertuples(index=True, name='Pandas'):
 print getattr(row, "c1"), getattr(row, "c2")

itertuples()应该比iterrows()快

但请注意,根据文档(目前 Pandas 0.19.1):

  • iterrows:数据的dtype可能不是按行匹配的,因为iterrows返回一个系列的每一行,它不会保留行的dtypes(dtypes跨DataFrames列保留)*
  • iterrows:不要修改行

你不应该修改你正在迭代的东西。这不能保证在所有情况下都能正常工作。根据数据类型的不同,迭代器返回一个副本而不是一个视图,写入它将不起作用。

改用DataFrame.apply():

new_df = df.apply(lambda x: x * 2)
itertuples:列名称将被重命名为位置名称,如果它们是无效的Python标识符,重复或以下划线开头。对于大量的列(> 255),返回常规元组。

第二种方案: apply

您也可以使用df.apply()遍历行并访问函数的多个列。

docs: DataFrame.apply()

def valuation_formula(x, y):
 return x * y * 0.5
 
df['price'] = df.apply(lambda row: valuation_formula(row['x'], row['y']), axis=1)

第三种方案:iloc

您可以使用df.iloc函数,如下所示:

for i in range(0, len(df)):
 print df.iloc[i]['c1'], df.iloc[i]['c2']

第四种方案:略麻烦,但是更高效,将DataFrame转为List

您可以编写自己的实现namedtuple的迭代器

from collections import namedtuple
 
def myiter(d, cols=None):
 if cols is None:
  v = d.values.tolist()
  cols = d.columns.values.tolist()
 else:
  j = [d.columns.get_loc(c) for c in cols]
  v = d.values[:, j].tolist()
 
 n = namedtuple('MyTuple', cols)
 
 for line in iter(v):
  yield n(*line)

这相当于pd.DataFrame.itertuples,但是效率更高。

将自定义函数用于给定的DataFrame:

list(myiter(df))
 
[MyTuple(c1=10, c2=100), MyTuple(c1=11, c2=110), MyTuple(c1=12, c2=120)]

或与pd.DataFrame.itertuples:

list(df.itertuples(index=False))
 
[Pandas(c1=10, c2=100), Pandas(c1=11, c2=110), Pandas(c1=12, c2=120)]

全面的测试

我们测试了所有可用列:

def iterfullA(d):
 return list(myiter(d))
 
def iterfullB(d):
 return list(d.itertuples(index=False))
 
def itersubA(d):
 return list(myiter(d, ['col3', 'col4', 'col5', 'col6', 'col7']))
 
def itersubB(d):
 return list(d[['col3', 'col4', 'col5', 'col6', 'col7']].itertuples(index=False))
 
res = pd.DataFrame(
 index=[10, 30, 100, 300, 1000, 3000, 10000, 30000],
 columns='iterfullA iterfullB itersubA itersubB'.split(),
 dtype=float
)
 
for i in res.index:
 d = pd.DataFrame(np.random.randint(10, size=(i, 10))).add_prefix('col')
 for j in res.columns:
  stmt = '{}(d)'.format(j)
  setp = 'from __main__ import d, {}'.format(j)
  res.at[i, j] = timeit(stmt, setp, number=100)
 
res.groupby(res.columns.str[4:-1], axis=1).plot(loglog=True);

在pandas中遍历DataFrame行的实现方法

在pandas中遍历DataFrame行的实现方法

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python使用cookie库操保存cookie详解
Mar 03 Python
python随机生成指定长度密码的方法
Apr 04 Python
Python编程中的文件操作攻略
Oct 16 Python
python3+PyQt5实现使用剪贴板做复制与粘帖示例
Jan 24 Python
python之PyMongo使用总结
May 26 Python
在Python web中实现验证码图片代码分享
Nov 09 Python
Python自定义函数定义,参数,调用代码解析
Dec 27 Python
Python处理文本换行符实例代码
Feb 03 Python
Python实现的读取/更改/写入xml文件操作示例
Aug 30 Python
Django框架实现分页显示内容的方法详解
May 10 Python
TensorFlow2.0:张量的合并与分割实例
Jan 19 Python
python爬虫--selenium模块
Mar 31 Python
python判断单向链表是否包括环,若包含则计算环入口的节点实例分析
Oct 23 #Python
Pandas DataFrame中的tuple元素遍历的实现
Oct 23 #Python
10行Python代码计算汽车数量的实现方法
Oct 23 #Python
在OpenCV里使用特征匹配和单映射变换的代码详解
Oct 23 #Python
手把手教你Python yLab的绘制折线图的画法
Oct 23 #Python
Python之Numpy的超实用基础详细教程
Oct 23 #Python
Python从列表推导到zip()函数的5种技巧总结
Oct 23 #Python
You might like
php设计模式 Decorator(装饰模式)
2011/06/26 PHP
php另类上传图片的方法(PHP用Socket上传图片)
2013/10/30 PHP
PHP生成可点击刷新的验证码简单示例
2016/05/13 PHP
PHP中一个有趣的preg_replace函数详解
2018/08/15 PHP
PHP与Perl之间知识点区别整理
2019/03/19 PHP
JavaScript入门教程(9) Document文档对象
2009/01/31 Javascript
JS 控制非法字符的输入代码
2009/12/04 Javascript
jQuery 美元符冲突的解决方法
2010/03/28 Javascript
php上传图片并给图片打上透明水印的代码
2010/06/07 Javascript
formvalidator验证插件中有关ajax验证问题
2013/01/04 Javascript
jquery实现商品拖动选择效果代码(自写)
2013/05/28 Javascript
把字符串按照特定的字母顺序进行排序的js代码
2014/01/28 Javascript
jQuery Trim去除字符串首尾空字符的实现方法说明
2014/02/11 Javascript
基于jQuery插件实现点击小图显示大图效果
2016/05/11 Javascript
jQuery.uploadify文件上传组件实例讲解
2016/09/23 Javascript
8 行 Node.js 代码实现代理服务器
2016/12/05 Javascript
vue2的todolist入门小项目的详细解析
2017/05/11 Javascript
详解从零搭建 vue2 vue-router2 webpack3 工程
2017/11/22 Javascript
vue实现的请求服务器端API接口示例
2019/05/25 Javascript
JS实现动态添加外部js、css到head标签的方法
2019/06/05 Javascript
JS实现吸顶特效
2020/01/08 Javascript
python对DICOM图像的读取方法详解
2017/07/17 Python
PyQt5打开文件对话框QFileDialog实例代码
2018/02/07 Python
python flask实现分页的示例代码
2018/08/02 Python
python 使用sys.stdin和fileinput读入标准输入的方法
2018/10/17 Python
Python利用Scrapy框架爬取豆瓣电影示例
2020/01/17 Python
Jupyter notebook无法导入第三方模块的解决方式
2020/04/15 Python
python3实现将json对象存入Redis以及数据的导入导出
2020/07/16 Python
Python基于内置函数type创建新类型
2020/10/22 Python
Python 打印自己设计的字体的实例讲解
2021/01/04 Python
详解CSS3原生支持div铺满浏览器的方法
2018/08/30 HTML / CSS
JavaScript获取当前url根目录(路径)
2014/02/19 面试题
2014第二批党员干部对照“四风”找差距检查材料思想汇报
2014/09/18 职场文书
学习退步检讨书
2014/09/28 职场文书
2016年寒假学习心得体会
2015/10/09 职场文书
V Rising 服务器搭建图文教程
2022/06/16 Servers