Группа панд с несколькими столбцами и максимальным значением

Question

У меня есть некоторые проблемы с group by с несколькими столбцами и значением max.

A   B   C   D   E   F   G   H

x   q   e   m   k   2   1   y
x   q   e   n   l   5   2   y
x   w   e   b   j   7   3   y
x   w   e   v   h   3   4   y

Этот запрос верный и возвращает то, что я хочу.

SELECT A, B, C, D, E, MAX(F) FROM mytable group by A, B, C

Результаты

 x   q   e   n   l   5
 x   w   e   b   j   7

Как это может быть достигнуто в пандах?

Я пытаюсь это:

df.groupby(['A', 'B', 'C'], as_index=False)['F'].max()

И это означает:

SELECT A, B, C, MAX(F) FROM mytable group by A, B, C

Это также не работает

df.groupby(['A', 'B', 'C'], as_index=False)['F','D','E'].max()

Как я могу вернуть также столбец D, E, как в SQL-запросе?

Answer 1

Похоже, вам нужно

groups = ['A', 'B', 'C']
selects = ['A', 'B', 'C','D', 'E','F']

df.groupby(groups, as_index=False).apply(lambda s: s.loc[s.F.idxmax(), selects]).reset_index(drop=True)

    A   B   C   D   E   F
0   x   q   e   n   l   5
1   x   w   e   b   j   7

Answer 2

Попробуйте что-то вроде этого:

df.groupby(['A', 'B', 'C'], as_index=False).agg({'D': 'first', 'E': 'last', 'F': 'max'})