[python] pandas 데이터에 함수 적용

출처 : 나도코딩 https://www.youtube.com/watch?v=PjhlUzp_cU0&t=10533

 

import pandas as pd

# excel 파일 열기
df = pd.read_excel('score.xlsx', index_col='지원번호')
print(df)

데이터에 함수 적용 : apply

데이터 중 ‘키’ column을 수정하려고 한다

지금은 데이터다 정수형의 값이지만, 문자열로 변환하여 ‘cm’를 붙이려고 한다

그럴 때 함수를 만들어서 apply를 통해 적용시킬 수 있다

def add_cm(height):
    return str(height)+'cm'

df['키'] = df['키'].apply(add_cm)
print(df)

이번에는 SW특기를 앞글자만 대문자로 변경해보도록 하자

그런데 SW특기에는 NaN데이터도 있기 때문에 pandas의 notnull을 사용해서 값이 있을떄만 진행해주자

def capitalize(lang):
    if pd.notnull(lang): # pandas 내장함수
        return lang.capitalize() # 내장함수

    return lang 

df['SW특기'] = df['SW특기'].apply(capitalize)
print(df)