Employee
表:
+-------------+------+ | Column Name | Type | +-------------+------+ | id | int | | salary | int | +-------------+------+ 在 SQL 中,id 是这个表的主键。 表的每一行包含员工的工资信息。
查询并返回 Employee
表中第二高的薪水 。如果不存在第二高的薪水,查询应该返回 null(Pandas 则返回 None)
。
解题思路同第N高薪水
employee = employee[['salary']].drop_duplicates()
if 2 > employee.shape[0]:
return pd.DataFrame({'SecondHighestSalary': [None]})
else:
employee = employee.sort_values(by='salary', ascending=False).head(2).tail(1)
# 改列名
employee = employee.rename(columns={'salary':'SecondHighestSalary'})
return employee