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

本文主要是介绍每日一题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 pddef second_highest_salary(employee: pd.DataFrame) -> pd.DataFrame:# 排序employee = employee.sort_values(by = 'salary', ascending = False)# 去除重复的工资employee = employee.drop_duplicates(subset = ['salary'])## 选择第二高的工资，如果不存在则返回nullif 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

这篇关于每日一题25：数据操作之第二高的薪水的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---