力扣：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 pd

def 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]})