本文主要是介绍力扣:177. 第N高的薪水(Python3),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题目:
表:
Employee
+-------------+------+ | Column Name | Type | +-------------+------+ | id | int | | salary | int | +-------------+------+ 在 SQL 中,id 是该表的主键。 该表的每一行都包含有关员工工资的信息。查询
Employee
表中第n
高的工资。如果没有第n
个最高工资,查询结果应该为null
。查询结果格式如下所示。
来源:力扣(LeetCode)
链接:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
示例:
示例 1:
输入:
Employee table: +----+--------+ | id | salary | +----+--------+ | 1 | 100 | | 2 | 200 | | 3 | 300 | +----+--------+ n = 2
输出:+------------------------+ | getNthHighestSalary(2) | +------------------------+ | 200 | +------------------------+
示例 2:输入:
Employee 表: +----+--------+ | id | salary | +----+--------+ | 1 | 100 | +----+--------+ n = 2
输出:+------------------------+ | getNthHighestSalary(2) | +------------------------+ | null | +------------------------+
解法:
思路类似力扣:176. 第二高的薪水(Python3)-CSDN博客,只需要注意2点:
修改iloc第1个下标,对应于行索引;
动态修改返回的列名。
代码:
import pandas as pddef nth_highest_salary(employee: pd.DataFrame, N: int) -> pd.DataFrame:try:return pd.DataFrame({'getNthHighestSalary(' + str(N) + ')': [employee.drop_duplicates(subset='salary').sort_values(['salary'], ascending=[False]).iloc[N - 1][1]]})except:return pd.DataFrame({'getNthHighestSalary(' + str(N) + ')': [None]})
这篇关于力扣:177. 第N高的薪水(Python3)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!