Средние цифровые столбцы и возвращайте отдельную строку для строковых столбцов с помощью Pandas Groupby ⇐ Python

[Anonymous]

У меня есть дата, которая выглядит как следующее: < /p>

Код: Выделить всё

    ID  Type    Size
0   123 Red     5
1   456 Blue    7
2   789 Yellow  12
3   789 Yellow  4

Я теперь хочу агрегировать по id и взять среднее значение для дубликатов. Тем не менее, я хочу вернуть только ту же строку для типа , а не объединить ее. Я попытался запечатлеть это, используя Agg :

Код: Выделить всё

df = pd.DataFrame({'ID' : [123, 456, 789, 789], 'Type' : ['Red', 'Blue', 'Yellow', 'Yellow'], 'Size' : [5, 7, 12, 4]})

def identity(x):
return x

special_columns = ['Type']
aggfuncs = {col: statistics.mean for col in df.columns}
aggfuncs.update({col:identity for col in special_columns})
df.groupby(['ID'], as_index=False).agg(aggfuncs)
< /code>
Однако это все еще превращается в массив повторной строки: < /p>
    ID  Type              Size
0   123 Red                 5
1   456 Blue                7
2   789 [Yellow, Yellow]    8
< /code>
конечный результат, который я хотел: < /p>
    ID  Type              Size
0   123 Red                 5
1   456 Blue                7
2   789 Yellow              8

Как это можно достичь?

Подробнее здесь: https://stackoverflow.com/questions/724 ... ing-pandas