logo标签列表

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

pythonpandasdataframegroup-bypandas-groupby
8

我有一个数据框，需要按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
1个回答
9
您可以先进行排序，然后再删除重复项：
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
网页内容由stack overflow 提供, 点击上面的
可以查看英文原文，
原文链接