Pandas Dataframe groupby：生成多个输出列

Question

我有以下代码：

def func(x):
  return (1, 2, 3)

df.groupby[col].aggregate(func)

如何通过一个聚合函数生成三列？我也尝试过返回np.array，pd.Series，但它没有帮助。

Answer 1

在func()中，您必须返回一个数据框，我相信您应该使用apply()，例如：{/ p>

def func(x):
    return pd.DataFrame([1,2,3]).T

df.groupby[col].apply(func)