570. 至少有5名直接下属的经理

570. 至少有5名直接下属的经理

2023年8月14日14:10:43

570. 至少有5名直接下属的经理

中等

76

相关企业

SQL Schema

---

Pandas Schema

---

表: Employee

+-------------+---------+
| Column Name | Type    |
+-------------+---------+
| id          | int     |
| name        | varchar |
| department  | varchar |
| managerId   | int     |
+-------------+---------+
在 SQL 中，id 是该表的主键列。
该表的每一行都表示雇员的名字、他们的部门和他们的经理的id。
如果managerId为空，则该员工没有经理。
没有员工会成为自己的管理者。

查询至少有5名直接下属的经理 。

以 任意顺序 返回结果表。

查询结果格式如下所示。

示例 1:

输入: 
Employee 表:
+-----+-------+------------+-----------+
| id  | name  | department | managerId |
+-----+-------+------------+-----------+
| 101 | John  | A          | None      |
| 102 | Dan   | A          | 101       |
| 103 | James | A          | 101       |
| 104 | Amy   | A          | 101       |
| 105 | Anne  | A          | 101       |
| 106 | Ron   | B          | 101       |
+-----+-------+------------+-----------+
输出: 
+------+
| name |
+------+
| John |
+------+

通过次数

42.4K

提交次数

67.1K

通过率

63.2%

答案

import pandas as pd

def find_managers(employee: pd.DataFrame) -> pd.DataFrame:
    gp = employee.groupby(by=['managerId'])
    size = gp.size()
    size = size[size >=5]
    df = employee[employee['id'].isin(size.index)]
    return df[['name']]


    if employee.empty:
        return employee[["name"]]
    temp = employee.groupby("managerId",as_index=False).agg({"name":lambda x:x.count()>=5})
    return employee[employee["id"].isin(temp[temp["name"]]["managerId"].tolist())][["name"]]
#    return  employee[employee.groupby("managerId",as_index=False).count()>=5].dropna(how='all')[["name"]]