[Python] 판다스 apply - 함수에 복수 인자 적용하기

목차

들어가기

Pandas DataFrame에서 apply 매소드를 사용해서 함수를 적용한다.

보통 단일 인자 함수를 사용하는데,

복수 인자를 넣을 수도 있다.

복수 인자를 사용하는 방식에는 두 가지가 있다.

1. 함수가 복수 인자를 받는 경우
2. 복수의 Column 의 데이터를 함수에 넘겨주는 경우

두 가지 방법에 대해서 알아보고 적절하게 사용하자.

 

복수 인자 사용하는 경우

import pandas as pd
df = pd.DataFrame({"A": [10, 20, 30], "B": [-10, -20, -30]})

def subtractAndMultiply(x, y, z):
    return (x - y) * z

df['A'].apply(subtractAndMultiply, args=(-10, 2))

Before DataFrame

df['A']의 값인 [10, 20, 30] 을 각각 substractAndMultiply 함수의 x 인자로 받는다.

그리고 args의 값들을 순서대로 y, z 값으로 받아서 함수 연산 결과를 출력한다.

After apply 적용

복수 Column의 데이터를 함수에 넘겨주는 경우

import pandas as pd
df = pd.DataFrame({"A": [10, 20, 30], "B": [-10, -20, -30]})

def subtract(x):
    return (x[0] - x[1])

df[['A', 'B']].apply(subtract, axis=1)

각 행의 A열과 B열의 데이터가 한 세트로 x 값에 들어가게된다.

After apply 적용 후

연산 방식의 필요에 따라서 적절하게 활용하면 좋다!