根据熊猫数据框中的另一列获取名称

时间:2021-02-05 01:38:58

标签: python pandas dataframe

我需要获取人力资源部门工作人员的姓名。


    FirstName   LastName        Year    Department
83  Joe         Faulk           2       Austin Public Library
84  Bryce       Benton          5       HR
85  Sarah       Cronin          7       Austin Public Library
86  Gabriel     Montgomery      2       Austin Resource Recovery
87  Patricia    Genty-Andrade   3       HR

这是我的代码,它显示了错误 AttributeError: 'DataFrame' object has no attribute 'unique'


names = df.iloc[:, 0:4][df['Department'] == 'HR'].unique()

我需要这样的输出

   FirstName   LastName        Department
0  Joe         Faulk           HR
1  Bryce       Benton          HR
2  Sarah       Cronin          HR
3  Gabriel     Montgomery      HR
4  Patricia    Genty-Andrade   HR

1 个答案:

答案 0 :(得分:0)

使用 drop_duplicates 而不是 unique,因为 unique 用于系列。

df.loc[df['Department'] == 'HR',
       ['FirstName', 'LastName', 'Department']].drop_duplicates()