每日一题25：数据操作之第二高的薪水

一、每日一题 

Employee 表：

+-------------+------+
| Column Name | Type |
+-------------+------+
| id          | int  |
| salary      | int  |
+-------------+------+
在 SQL 中，id 是这个表的主键。
表的每一行包含员工的工资信息。

查询并返回 Employee 表中第二高的薪水 。如果不存在第二高的薪水，查询应该返回 null(Pandas 则返回 None) 。

查询结果如下例所示。

示例 1：

输入：
Employee 表：
+----+--------+
| id | salary |
+----+--------+
| 1  | 100    |
| 2  | 200    |
| 3  | 300    |
+----+--------+
输出：
+---------------------+
| SecondHighestSalary |
+---------------------+
| 200                 |
+---------------------+

示例 2：

输入：
Employee 表：
+----+--------+
| id | salary |
+----+--------+
| 1  | 100    |
+----+--------+
输出：
+---------------------+
| SecondHighestSalary |
+---------------------+
| null                |
+---------------------+

解答： 

import pandas as pd

def second_highest_salary(employee: pd.DataFrame) -> pd.DataFrame:
    # 排序
    employee = employee.sort_values(by = 'salary', ascending = False)
    # 去除重复的工资
    employee = employee.drop_duplicates(subset = ['salary'])
    ## 选择第二高的工资，如果不存在则返回null
    if len(employee) >= 2:
        return pd.DataFrame({'SecondHighestSalary':[int(employee.iloc[1]['salary'])]})
    else:
        return pd.DataFrame({'SecondHighestSalary': [pd.NA]})

题源：Leetcode

二、总结

该题和昨天的题类似，甚至说更简单。这里值得注意的点是[pd.NA],而不能写成['null']

2024.5.28