Панды - группировка, сортировка и сохранение первого ряда

Question

У меня есть фрейм данных, который мне нужно сгруппировать по id, затем отсортировать по time и сохранить только первую запись для каждого id.Пробовал разные методы, предложенные в других ответах, но не смог получить правильный результат.Будем благодарны за помощь!

test = pd.DataFrame({'id' : [1,1,1,
                           2,2,
                           3,3,3,3],
                   'ref'  : ["search","social","direct",
                          "social","direct",
                          "direct",'social','social','social'],
                   'media':['video', 'page', 'video',
                           'page', 'pic', 
                            'pic', 'video', 'page', 'video'],
                   'time': ['2019-04-10 19:00:00.569', '2019-04-10 00:10:00.569', '2019-04-10 00:10:20.569',
                          '2019-04-10 12:10:00.569','2019-04-10 11:10:00.569',
                          '2019-04-10 22:10:00.569','2019-04-10 14:10:00.569','2019-04-10 14:30:00.569','2019-04-10 15:10:00.569']})

Ожидаемый результат:

    id  ref     media
0   1   social  page
1   2   direct  pic
2   3   social  video

Answer 1

Вы можете отсортировать, а затем удалить дубликаты:

test.sort_values(by=['id', 'time']).drop_duplicates('id').drop('time',1)

   id     ref  media
1   1  social   page
4   2  direct    pic
6   3  social  video