Сохраняйте нечисловые столбцы при выполнении pandas.DataFrame.groupby (). Sum ()

Группировка только по столбцу «А» дает:

df.groupby(['A']).sum()

        C     D
A              
bar  1.26  0.88
foo  0.92 -4.19

Группировка по столбцам «А» и «В» дает:

df.groupby(['A','B']).sum()

            C     D
A   B                
bar one    1.38 -0.73
    three  0.26  0.80
    two   -0.38  0.81
foo one    1.96 -2.72
    three -0.42 -0.18
    two   -0.62 -1.29

Если вы хотите только столбец «B», который имеет «один», вы можете сделать:

d = df.groupby(['A','B'], as_index=False).sum()
d[d.B=='one'].set_index('A')

    B     C     D
A                   
bar  one  1.38 -0.73
foo  one  1.96 -2.72

Я не уверен, что понимаю, но это то, что вы хотите сделать? Примечание: я увеличил точность вывода только для того, чтобы получить те же цифры, что и в посте.

d = df.groupby('A').sum()
d['B'] = 'one'
d.sort_index(axis=1)

       B         C         D
A                           
bar  one  1.259069  0.876959
foo  one  0.921510 -4.193397

Если вы хотите поместить первое отсортированное значение столбца из 'B', вы можете использовать:

d['B'] = df.B.sort(inplace=False)[0]

Итак, здесь я заменил «один», «два», «три» на «a», «b», «c», чтобы увидеть, действительно ли это то, что вы пытаетесь сделать, и использовал метод insert (), как было предложено в другом посте.

df

    A  B         C         D
0  foo  a  0.638362 -0.931817
1  bar  a  1.380706 -0.733307
2  foo  b -0.324514  0.203515
3  bar  c  0.258534  0.803298
4  foo  b -0.299485 -1.495979
5  bar  b -0.380171  0.806968
6  foo  a  1.324810 -1.792996
7  foo  c -0.417663 -0.176120

d = df.groupby('A').sum()
d.insert(0, 'B', df.B.sort(inplace=False)[0])
d

    B         C         D
A                         
bar  a  1.259069  0.876959
foo  a  0.921510 -4.193397

Просто используйте «сначала»:

df.groupby(["A"]).agg({'B': 'first',
                       'C': 'first',
                       'D': sum,
                       'E': sum})

Все эти ответы кажутся довольно многословными; Документ Pandas [1] также не совсем ясен по этому вопросу, несмотря на то, что на первой странице приведен пример SQL:

SELECT Column1, Column2, mean(Column3), sum(Column4)
FROM SomeTable
GROUP BY Column1, Column2

Который выберет «первое» из Column1, Column2, а Pandas - нет. Как указывает OP, нечисловые столбцы просто удаляются. Мое решение тоже не очень красивое, но, возможно, больше контроля - цель дизайна Pandas:

agg_d = { c: 'sum' if c == 'A' else 'first' for c in df.columns }
df = df.groupby( df['A'] ).agg( agg_d )

Это поддерживает все нечисловые столбцы, как это делает sql. Это в основном совпадает с ответом Филиппа ниже, но без необходимости явно перечислять столбцы.

ПРИМЕЧАНИЯ

1. https://pandas.pydata.org/pandas-docs/stable/groupby.html