pandas groupby，并设置项目集

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

更新资料

• 直到大熊猫0.22版，这才成为问题。
• 从熊猫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'>

根据文档，.agg()期望：

arg：function或dict

用于汇总组的功能。

• 如果为a function，则必须在传递aDataFrame或传递给时 起作用DataFrame.apply。
  
  

• 如果传递了dict，则键必须是DataFrame列名。
  
  

可接受的组合是：

• string cythonized函数名称

• function
  
  

• list 功能的
  

• dict 列数->函数
  

• dict名称嵌套->函数字典