Create a New Column (#05)

import pandas as pd

def createBonusColumn(employees: pd.DataFrame) -> pd.DataFrame:
    employees['bonus'] = employees['salary'] * 2
    return employees 

---

new column

df.insert(loc, column, value, allow_duplicates=False)

• 특정 위치에 열(column)을 삽입하는 메서드

loc은 행 번호, columns은 칼럼명, value는 (보통) list이다.
allow_duplicates를 True로 하면, 열 이름이 같더라도(중복되더라도) 추가시킨다.

df.assign(칼럼명1=값1, 칼럼명2=값2)

• 새 열(column)을 할당하는 메서드

---

<다른 풀이>

import pandas as pd

def createBonusColumn(employees: pd.DataFrame) -> pd.DataFrame:
    return employees.assign(bonus = lambda x: x.salary * 2)
    #여기서 x는 employees임.

lambda

lambda 매개변수(입력값): 표현식(반환값)
lambda는 익명 함수(Anonymous Function)를 만들 때 사용하는 Python의 내장 키워드이다. 즉, 이름 없이 간결하게 정의하는 함수로, 짧고 간단한 연산을 수행할 때 사용한다.

#예시
add = lambda x, y: x + y
print(add(3, 5))  # 출력: 8