Pandas.DataFrame实现分组、排序并且为分组插入排名

最新推荐文章于 2024-03-03 15:50:25 发布

GYT0313

最新推荐文章于 2024-03-03 15:50:25 发布

阅读量8.1k

点赞数 11

分类专栏： Python Python数据分析及可视化文章标签： Pandas DataFrame groupby sort_values

本文链接：https://blog.csdn.net/qq_38038143/article/details/102624344

版权

Python 同时被 2 个专栏收录

55 篇文章 5 订阅

订阅专栏

Python数据分析及可视化

8 篇文章 0 订阅

订阅专栏

1. 示例数据（各班级学生得分）

import pandas as pd
data_dict = {"name": 
             ["Rose", "Jack", "Tom", "Kyle", "Jery", "Adam", "Bill", "David", "Denny", "Evan"],
            "class": [1, 2, 3, 1, 2, 3, 1, 2, 3, 1],
            "score": [88, 92, 38, 98, 22, 65, 45, 53, 97, 100]}
df = pd.DataFrame(data=data_dict)
df

在这里插入图片描述

2. 按班级分组

df = df.groupby('class', sort=False)\
        .apply(lambda x:x.sort_values("score", ascending=False))\
        .reset_index(drop=True)
df

在这里插入图片描述

3. 给各分组班级增加排名列

df["rank"] = None
# 标识班级
flag = df.loc[0].values[1]
rank = 0
for i in range(len(df)):
    temp = df.loc[i].values[1]
    if (temp == flag).all():
        # 同一班级
        rank += 1
    else:
        # 不同班级，重新计算排名
        flag = temp
        rank = 1
    df.loc[i, "rank"] = rank
df