[EDA/Python] Playing with Pandas 📊🐼 2편

Applying Functions to Dataframes

사실 필자는 영어가 더 편하다. 필자의 학습을 위해 작성 중인 블로그인 만큼 한국말이 생각나지 않으면 그냥 영어로 적도록 하겠다. 🐼

apply()를 활용하여 df 혹은 Series에 함수를 적용시킬 수 있다.

"Dataframe C"

display(C)
G = C.copy() #Copy를 하지 않으면 the dataframe is modified 
G['year] = G['year'].apply(lambda x: "'{:02d}".format(x % 100)) 
display(G) 

"Dataframe G"

요렇게 쓴다 이말이야. 근데 이게 상당히 복잡하고 유용해진다.

두 열에 대한 연산을 통해 새로운 열 생성하기

우선 axis = 0과 axis = 1의 방향을 잊지 말자.

G['prevalence'] = G['cases'] / G['popuation'] 

물론 위가 가장 간단한 방법이지만 apply함수를 활용하는 함수를 작성해보자.

def calc_prevalence(G):
    assert 'cases' in G.columns and 'population' in G.columns
    F = G.copy()
    F['prevalence'] = F.apply(lambda row : row['cases']/row['population'], axis=1)
    
    return F
display(calc_prevalence(G))