繁体   English   中英

大熊猫行具体适用

[英]pandas row specific apply

此R问题类似,我想使用Pandas将函数应用于Series(或DataFrame中的每一行)中的每个项目,但是希望将该行的索引或id用作此函数的参数。 作为一个简单的例子,假设有人想要创建[(index_i,value_i),...,(index_n,value_n)]形式的元组列表。 使用简单的Python for循环,我可以这样做:

In [1] L = []
In [2] s = Series(['six', 'seven', 'six', 'seven', 'six'],
           index=['a', 'b', 'c', 'd', 'e'])
In [3] for i, item in enumerate(s):
           L.append((i,item))
In [4] L
Out[4] [(0, 'six'), (1, 'seven'), (2, 'six'), (3, 'seven'), (4, 'six')]

但必须有一个更有效的方法来做到这一点? 或许更像Panda-likeh喜欢Series.apply? 实际上,我并不担心(在这种情况下)返回任何有意义的东西,但更多的是为了“应用”之类的效率。 有任何想法吗?

如果对函数使用apply方法,那么系列中的每个项都将使用这样的函数进行映射。 例如

>>> s.apply(enumerate)
a    <enumerate object at 0x13cf910>
b    <enumerate object at 0x13cf870>
c    <enumerate object at 0x13cf820>
d    <enumerate object at 0x13cf7d0>
e    <enumerate object at 0x13ecdc0>

你想要做的只是枚举系列本身。

>>> list(enumerate(s))
[(0, 'six'), (1, 'seven'), (2, 'six'), (3, 'seven'), (4, 'six')]

如果您想要对所有实体的字符串求和,该怎么办?

>>> ",".join(s)
'six,seven,six,seven,six'

申请的更复杂用法是:

>>> from functools import partial
>>> s.apply(partial(map, lambda x: x*2 ))
a                ['ss', 'ii', 'xx']
b    ['ss', 'ee', 'vv', 'ee', 'nn']
c                ['ss', 'ii', 'xx']
d    ['ss', 'ee', 'vv', 'ee', 'nn']
e                ['ss', 'ii', 'xx']

[编辑]

根据OP的澄清问题:不要将系列(1D)与DataFrames(2D) http://pandas.pydata.org/pandas-docs/stable/dsintro.html#dataframe混淆 - 因为我没有真正看到你可以谈论行。 但是,您可以通过创建新系列在函数中包含索引(应用不会为您提供有关当前索引的任何信息):

>>> Series([s[x]+" my index is:  "+x for x in s.keys()], index=s.keys())
a      six index  a
b    seven index  b
c      six index  c
d    seven index  d
e      six index  e

无论如何,我建议你切换到其他数据类型,以避免巨大的内存泄漏。

这是一个简洁的方法,使用itertools的countzip

import pandas as pd
from itertools import count

s = pd.Series(['six', 'seven', 'six', 'seven', 'six'],
                  index=['a', 'b', 'c', 'd', 'e'])

In [4]: zip(count(), s)
Out[4]: [(0, 'six'), (1, 'seven'), (2, 'six'), (3, 'seven'), (4, 'six')]

不幸的是,只有enumerate(list(s))才有效!

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM