콘솔워크

Python 데이터프레임 두 컬럼 하나로 합치기 (dataframe combine two columns) 본문

프로그래밍/python

Python 데이터프레임 두 컬럼 하나로 합치기 (dataframe combine two columns)

콘솔워크 2021. 8. 23. 16:08
반응형

python dataframe의 two column들을 특정 문자열을 붙여서 하나로 합치려고 한다.

기본 식은 간단하다.

df['result'] = df['col1'] + " " + df['col2']

 

이렇게 수식처럼 두개를 붙여주면 된다.

 

하지만 나에게 문제는 df['col2']의 값이 없을 때는 두 개가 제대로 안 합쳐지는 결과가 나왔다.

 

이에 lamda 함수를 붙여서 합쳐주는 것으로 변경하였다.

 

def combine_2rd_columns(col_1, col_2):
    result = col_1
    if not pd.isna(col_2):
        result += " " + str(col_2)
    return result

df["result"] = df.apply(lambda x: combine_2rd_columns(x['col_1'], x['col_2']), axis=1)

 

상황에 맞게 정의한 함수를 바꾸면 된다.

 

 

 

반응형