python - pandas框架下如何减少一个字符串最长的数据？

Question

如何在pandas框架下归约一个字符串最长的数据？我尝试了以下代码，但得到了 ValueError: invalid number of arguments。

def f1(s):
    return max(s, key=len)

data.groupby('id').agg({'name':(lambda s: f1(s)) })

例。输入

id name
GB "United Kingdom"
GB England
US "United States"
US America

输出:

id name
GB "United Kingdom"
US "United States"

Answer 1

代码应该可以工作。顺便说一句，您不需要将 f1 包装在另一个 lambda 中。只需传递 f1。 (它们具有完全相同的参数签名)

>>> import pandas as pd
>>>
>>> def f1(s):
...     return max(s, key=len)
...
>>> data = pd.DataFrame([
...     {'id': 'GB', 'name': '"United Kingdom"'},
...     {'id': 'GB', 'name': 'England'},
...     {'id': 'US', 'name': '"United States"'},
...     {'id': 'US', 'name': 'America'},
...
... ])
>>> data.groupby('id').agg({'name': f1})
                name
id
GB  "United Kingdom"
US   "United States"