Панды Dataframe GroupBy: производят несколько столбцов выходных

Question

У меня есть следующий код:

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

df.groupby[col].aggregate(func) 

Как сделать три столбца в качестве результата функции агрегации один? Я также попытался вернуть np.array, pd.Series, но это не поможет.

Answer 1

В func() вы должны вернуть dataframe, и я считаю, вы должны использовать apply(), например так:

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

df.groupby[col].apply(func)