[python] 파이썬 pandas에서 데이터프레임의 열을 조건에 따라 변환하는 방법은 무엇인가요?

20 Dec 2023

python

우선, 다음과 같은 데이터프레임을 가정합니다.

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)
print(df)

# 결과
#    A   B
# 0  1  10
# 1  2  20
# 2  3  30
# 3  4  40
# 4  5  50

이제 ‘A’ 열의 값이 3보다 작으면 해당 값을 0으로, 3보다 크거나 같으면 해당 값을 1로 변환하는 방법은 다음과 같습니다.

df['A'] = df['A'].apply(lambda x: 0 if x < 3 else 1)
print(df)

# 결과
#    A   B
# 0  0  10
# 1  0  20
# 2  1  30
# 3  1  40
# 4  1  50

위 코드에서는 apply 함수와 lambda 함수를 사용하여 ‘A’ 열의 값을 조건에 따라 변환했습니다.

위의 예제는 간단한 조건에 따른 변환을 보여주는 예제이며, 복잡한 조건이나 다양한 변환 로직을 적용할 수도 있습니다.