[leetcode] 175. Combine Two Tables

테이블 생성코드

 

data = [[1, 'Wang', 'Allen'], [2, 'Alice', 'Bob']]person = pd.DataFrame(data, columns=['personId', 'firstName', 'lastName']).astype({'personId':'Int64', 'firstName':'object', 'lastName':'object'})data = [[1, 2, 'New York City', 'New York'], [2, 3, 'Leetcode', 'California']]address = pd.DataFrame(data, columns=['addressId', 'personId', 'city', 'state']).astype({'addressId':'Int64', 'personId':'Int64', 'city':'object', 'state':'object'})

 

문제 

 

 

해결 코드

import pandas as pd

def combine_two_tables(person: pd.DataFrame, address: pd.DataFrame) -> pd.DataFrame:

    merge_df = pd.merge(person,address,how='left')

    combine_two_tables = merge_df[['firstName','lastName','city','state']]

    return combine_two_tables

 

 

풀이

merge 를 사용하는데 인자로 합치고 싶은 두 테이블의 이름과 how='left' 를 넣어서 left join과 같은 효과가 나도록 했다.

합친 데이터 프레임을 merge_df 에 저장하고 merge_df 에서 출력하고 싶은 컬럼만 다시 지정해서 combine_two_tables 에 넣었다.

그리고 return  combine_two_tables 해주었다!