【Python】pandas实现row_number() over()函数

Pandas分组排序技巧

最新推荐文章于 2025-06-28 22:00:07 发布

转载最新推荐文章于 2025-06-28 22:00:07 发布 · 1.3w 阅读

文章标签：

#Python

编程专栏收录该内容

25 篇文章

订阅专栏

本文介绍了一种使用Pandas库进行数据集分组并排序的方法。通过定义函数row_number，可以实现DataFrame格式数据集的分组依据字段、排序依据字段的设定，以及升序或降序的选择。此方法利用了Pandas的groupby和rank函数，适用于需要对数据进行复杂排序处理的场景。

原文：

https://blog.youkuaiyun.com/qq_22238533/article/details/72395564

'''
dataset: DataFrame格式数据集
partionby：分组依据字段
orderby：排序依据字段
asc:是否为升序；1:升序；0:降序
return series格式：序号
'''
def row_number(dataset, partionby, orderby, asc):
    return dataset[orderby].groupby(dataset[partionby]).rank(ascending=asc, method='first')