python - Pandas - 分组、排序并保留第一行

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