查找一列具有 2 个特定值的所有 id

Question

我有一个 dataframe，其中包含员工、他们的经理和他们从事的项目的数据。 dataframe（有点简化）看起来像这样：

    EmployeeID  ManagerID  ProjectID
0            2         18        111
1            3         17        111
2            2         17        119
3            3         22        121
4            6         22        121
5            6         18        111
6            6         17        113
7            6         17        116

我想要所有与经理 17 和 18 一起工作的员工，在这种情况下是员工 2 和员工 6。

我知道我可以编写一个查询来让所有与经理 17 或 18 一起工作的员工使用：

df.query('ManagerID == 17 | ManagerID == 18')

但是现在我需要找到所有使用过机器人的员工，因为可以在 Z6A8064B5DF4794555500553C47C55057DZ 中多次找到员工 - 经理的组合，我无法使用计数。 我想我需要一个自我加入，但我真的不知道如何在 pandas 中完成。

Answer 1

您可以将DataFrame.drop_duplicates与DataFrame.pivot和DataFrame.dropna的所有EmployeeID一起使用。

df = df.query('ManagerID == 17 | ManagerID == 18')
#another solution for filter 
#df = df.query('ManagerID in [17, 18]')
emp = (df.drop_duplicates(subset=['EmployeeID','ManagerID'])
        .pivot('EmployeeID','ManagerID','ProjectID')
        .dropna()
        .index
        .tolist())
print (emp)
[2, 6]