相关文章推荐
八块腹肌的遥控器  ·  大卫·芬奇(美国男导演、编剧)_搜狗百科·  2 年前    · 
朝气蓬勃的石榴  ·  上海中医药大学、南京中医药大学等12所地方“ ...·  2 年前    · 
没有腹肌的胡萝卜  ·  轮船怎么画?教你绘制简易轮船的画法教学! - 知乎·  2 年前    · 
乐观的黄花菜  ·  用户满意你的产品吗?这14个用户调查问题给你 ...·  2 年前    · 
刚分手的鸡蛋  ·  地海传说(美国2004年Robert ...·  2 年前    · 
Code  ›  如何遍历pandas当中dataframe的行开发者社区
社区功能 pandas dataframe 遍历
https://cloud.tencent.com/developer/article/1451643
阳刚的红茶
2 年前
作者头像
马哥Python
0 篇文章

如何遍历pandas当中dataframe的行

前往专栏
腾讯云
开发者社区
文档 意见反馈 控制台
首页
学习
活动
专区
工具
TVP
文章/答案/技术大牛
发布
首页
学习
活动
专区
工具
TVP
返回腾讯云官网
社区首页 > 专栏 > 快乐学Python > 如何遍历pandas当中dataframe的行

如何遍历pandas当中dataframe的行

作者头像
马哥Python
发布 于 2019-06-27 16:52:51
2.5K 0
发布 于 2019-06-27 16:52:51
举报

有如下 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:
 
推荐文章
八块腹肌的遥控器  ·  大卫·芬奇(美国男导演、编剧)_搜狗百科
2 年前
朝气蓬勃的石榴  ·  上海中医药大学、南京中医药大学等12所地方“中医药大学”,哪个实力 ...
2 年前
没有腹肌的胡萝卜  ·  轮船怎么画?教你绘制简易轮船的画法教学! - 知乎
2 年前
乐观的黄花菜  ·  用户满意你的产品吗?这14个用户调查问题给你答案| 人人都是产品经理
2 年前
刚分手的鸡蛋  ·  地海传说(美国2004年Robert Lieberman执导电影)_搜狗百科
2 年前
今天看啥   ·   Py中国   ·   codingpro   ·   小百科   ·   link之家   ·   卧龙AI搜索
删除内容请联系邮箱 2879853325@qq.com
Code - 代码工具平台
© 2024 ~ 沪ICP备11025650号