[Python] 판다스 두 컬럼의 값을 비교하여 True/False 컬럼 만들기

다음과 같은 데이터프레임이 있다.

import pandas as pd

first_data  = [i for i in range(10, 20)]
second_data = 15

df = pd.DataFrame({'first': first_data, 
                   'second': second_data})

print(df)

 

first의 값이 second 보다 크면 True 를 출력하고 아니면 False 를 출력하는 result 행을 만들어보자

import pandas as pd

first_data  = [i for i in range(10, 20)]
second_data = 15

df = pd.DataFrame({'first': first_data, 
                   'second': second_data})

df['result'] = df['first'] > df['second']

print(df)

 

부호를 반대로 넣어보자

import pandas as pd

first_data  = [i for i in range(10, 20)]
second_data = 15

df = pd.DataFrame({'first': first_data, 
                   'second': second_data})

df['result'] = df['first'] < df['second']

print(df)

 

False 값을 특정 값으로 변경해보기

import pandas as pd

first_data  = [i for i in range(10, 20)]
second_data = 15

df = pd.DataFrame({'first': first_data, 
                   'second': second_data})

df['result'] = df['first'] < df['second']

df['result'] = df['result'].replace(False, 'Oh My God')

print(df)

 

두 값을 비교해서 문자열을 넣어보자

import pandas as pd

first_data  = [i for i in range(10, 20)]
second_data = 15

df = pd.DataFrame({'first': first_data, 
                   'second': second_data})

df['result'] = df['first'] < df['second']

df.loc[df['first'] > df['second'], 'result'] = df['first']
df.loc[df['first'] < df['second'], 'result'] = df['second']
df.loc[df['first'] == df['second'], 'result'] = 'equal'

print(df)