用户定义的功能不适用于Pandas

金属牧师

我正在学习Python Pandas,因此编写了一些代码来尝试与一起使用的用户定义函数agg,如下所示。

import pandas as pd

def combine_cities(series):
       return reduce(lambda x, y: x + ', ' + y, series)

data = pd.DataFrame({'Country': ['Russia','USA','China','USA','China'],
                    'City':['Moscow','Boston','Wuhan','New York','Beijing']})
a = data.groupby('Country').agg(combine_cities)
print(a)

但是,出现以下错误。知道我在做什么错吗?

NameError                                 Traceback (most recent call last)
~\Anaconda3\lib\site-packages\pandas\core\groupby\ops.py in agg_series(self, obj, func)
    662         try:
--> 663             return self._aggregate_series_fast(obj, func)
    664         except Exception:

~\Anaconda3\lib\site-packages\pandas\core\groupby\ops.py in _aggregate_series_fast(self, obj, func)
    680         grouper = reduction.SeriesGrouper(obj, func, group_index, ngroups, dummy)
--> 681         result, counts = grouper.get_result()
    682         return result, counts

pandas\_libs\reduction.pyx in pandas._libs.reduction.SeriesGrouper.get_result()

pandas\_libs\reduction.pyx in pandas._libs.reduction.SeriesGrouper.get_result()

....
金属牧师

发生错误的原因是Python3删除了reduce函数。因此,我必须添加以下内容才能使其正常工作。

from functools import reduce

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章