pandas笔记（四）-- 第N高的薪水（去重、排序、含空值的数据表）

合集 - pandas随笔(7)

1.pandas笔记（六）-- 即时食物配送2024-07-252.pandas笔记（五）-- 部门工资最高的员工（数据表的合并与分组）2024-03-27

3.pandas笔记（四）-- 第N高的薪水（去重、排序、含空值的数据表）2024-03-21

4.pandas笔记（三）-- 查找有效邮箱的用户（正则表达式应用）2024-03-085.pandas笔记（二）-- 从不订购的顾客 （数据表连接，主键与外键）2024-03-066.pandas笔记（一）-- 大的国家（逻辑索引、切片）2024-03-067.pandas随笔（七）-- 统计文本中单词出现的次数（词频统计）2024-10-17

收起

目录

• 题目描述
• 测试用例
• 解析
  • 思路
  • 解题方法
  • 代码

题目描述

查询employee表中第N高的工资，如果没有第n高的工资则返回null
如下所示：

测试用例

输入

id	salary
1	100
2	200
3	300

n=2

输出

getNthHighestSalary(2)

200

输入

id	salary
1	100
2	200
3	300

n=4

输出

getNthHighestSalary(2)

null

解析

思路

讲述看到这一题的思路

解题方法

首先需要除去employee表中工资相同的行，再按照去重后的数据表进行查找，即
employee = employee.drop_duplicates(subset=["salary"])，随后判断
n值的范围是否超出数据表的索引范围，即if n > len(employee) or n <= 0
若超出则返回列索引为f"getNthHighestSalary({n})"的空表，若未超出则
进行排序、查找与返回操作，注意为返回的数据表进行重命名

代码

import pandas as pd

def nth_highest_salary(employee: pd.DataFrame, n: int) -> pd.DataFrame:
    employee = employee.drop_duplicates(subset=["salary"])
    if n > len(employee) or n <= 0:
        return pd.DataFrame({f"getNthHighestSalary({n})": [None]})
    df = employee.sort_values(by="salary", ascending=False)
    return df[["salary"]].iloc[[n-1]].rename(columns={"salary": f"getNthHighestSalary({n})"})