python - Pandas 数据框 groupby : produce multiple output columns

Question

我有以下代码:

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

df.groupby[col].aggregate(func)

如何将三列作为一个聚合函数的结果？我也尝试返回 np.array、pd.Series，但没有帮助。

Answer 1

在 func() 中，您必须返回一个数据帧，我相信您应该使用 apply()，例如如下所示:

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

df.groupby[col].apply(func)