在 SQL 中,我们可以使用 JOIN
语句将员工表与自身连接,以便比较每个员工的工资和其经理的工资。以下是 SQL 解决方案:
SQL 方案
SELECT e1.name AS Employee
FROM Employee e1
JOIN Employee e2 ON e1.managerId = e2.id
WHERE e1.salary > e2.salary;
解释:
e1
代表员工,e2
代表经理,我们通过e1.managerId = e2.id
连接表格。WHERE e1.salary > e2.salary
用于筛选工资比经理高的员工。
Pandas 方案
在 Pandas 中,我们可以使用 merge
进行自连接,并通过布尔索引筛选符合条件的员工。
import pandas as pd
# 示例数据
data = {
"id": [1, 2, 3, 4],
"name": ["Alice", "Bob", "Charlie", "David"],
"salary": [5000, 7000, 6000, 8000],
"managerId": [None, 1, 1, 2]
}
df = pd.DataFrame(data)
# 进行自连接,将员工与其经理的数据合并
merged_df = df.merge(df, left_on="managerId", right_on="id", suffixes=('', '_manager'))
# 筛选工资比经理高的员工
result = merged_df[merged_df["salary"] > merged_df["salary_manager"]][["name"]]
# 重命名列
result.columns = ["Employee"]
# 输出结果
print(result)
解释:
- 通过
merge
将df
自连接,使得员工与其经理匹配。 - 通过
merged_df["salary"] > merged_df["salary_manager"]
筛选出符合条件的行。 - 选择
name
列并重命名为Employee
以符合要求。