python – Pandas groupby并制作一组项目

我正在使用pandas groupby并希望应用该功能从组中的项目创建一个集合.

以下不起作用：

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})

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

解决方法:

这是因为set是类型类型而to_set是类型函数：

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 to
  DataFrame.apply.

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

Accepted Combinations are:

• string cythonized function name

• function

• list of functions

• dict of columns -> functions

• nested dict of names -> dicts of functions