p181

choco·2023년 11월 18일

pandas

목록 보기
2/3

merge 시, 칼럼이름이 다를 때,

left_on, right_on 을 사용

employee = pd.merge(left = employee, right = employee, how= 'left', left_on = "managerId", right_on = "id")

두 개 이상의 열을 가져올 때

df [["칼럼이름", "칼럼이름"]] 의 방법을 사용. dataframe형태로 가져오기 위해 사용

column rename

딕셔너리 사용

employee.rename(columns={'name_x' : 'Employee'})

Code

def find_employees(employee: pd.DataFrame) -> pd.DataFrame:
    employee = pd.merge(left = employee, right = employee, how= 'left', left_on = "managerId", right_on = "id")
    employee = employee[employee["salary_x"] > employee["salary_y"]][['name_x']]
    return employee.rename(columns={'name_x' : 'Employee'})
profile
choco_lee

0개의 댓글