Как получить первую группу в группе из нескольких столбцов? ⇐ Python

[Anonymous]

Я пытался понять, как можно вернуть только первую группу после применения groupby.

Мой код выглядит так:

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

gb = df.groupby(['col1', 'col2', 'col3', 'col4'])['col5'].sum()

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

Вот пример моего результата:

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

col1  col2  col3   col4  'sum'
1     34   green   10    0.0
yellow  30    1.5
orange  20    1.1
2     89   green   10    3.0
yellow   5    0.0
orange  10    1.0

Я хочу, чтобы мне вернули следующее:

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

col1  col2  col3   col4  'sum'
1     34   green   10    0.0
yellow  30    1.5
orange  20    1.1

(Обратите внимание на столбец «сумма», который я только что добавил сюда, чтобы было понятно, что это был за последний столбец, но pandas на самом деле не называет этот столбец)

Подробнее здесь: https://stackoverflow.com/questions/497 ... le-columns