python - Pandas 分组并制作一组项目

Question

我正在使用 pandas groupby 并希望应用该函数从组中的项目中创建一个集合。
以下结果在 TypeError: 'type' object is not iterable :

df = df.groupby('col1')['col2'].agg({'size': len, 'set': set})

但以下工作:

def to_set(x):
    return set(x)
    
df = df.groupby('col1')['col2'].agg({'size': len, 'set': to_set})

在我的理解中，这两个表达式是相似的，第一个不起作用的原因是什么？

Answer 1

更新

迟至 Pandas 0.22 版，这是一个问题。

从 pandas 版本 1.1.2 开始，这不是问题。聚合 set , 不会导致 TypeError: 'type' object is not iterable .

不确定功能何时更新。

原答案
因为 set是 type type而 to_set是 type function :

type(set)
<class 'type'>

def to_set(x):
    return set(x)

type(to_set)

<class 'function'>

根据 docs , .agg()期望:

arg : function or dict

Function to use for aggregating groups.

• If a function, must either work when passed a DataFrame or when passed toDataFrame.apply.

• If passed a dict, the keys must be DataFrame columnnames.

Accepted Combinations are:

• string cythonized function name
• function

• list of functions

• dict of columns -> functions

• nested dict of names -> dicts of functions