[Pandas] 다른 데이터프레임 속성값을 이용해 값 선택, 데이터프레임 순회

두 개의 데이터프레임이 있을 때, 한 쪽 데이터프레임의 속성값을 사용해 다른 데이터 프레임에서 행을 선택하는 방법에 대해 적어보려 합니다.

 

다음과 같이 전체 매출 DataFrame과 반품-취소 DataFrame이 있습니다. 반품-취소 DataFrame을 사용해 전체 매출 데이터에서 해당하는 행을 얻으려 합니다.

전체 매출 데이터, sales

반품-취소 데이터프레임, cancel_list

import pandas as pd

selected_rows = []

for idx, item in cancel_list.iterrows():
    selected_rows.append(sales.loc[(sales['판매일자'] == item['판매일자']) & (sales['매출금액'] == item['판매일자'])])
    
selected_rows

DataFrame의 iterrows 메소드는 DataFrame을 순환하며 각 행의 인덱스와 해당하는 속성값들을 Series로 반환합니다.

이외에도 데이터프레임을 순환하는 경우 유용하게 사용할 수 있는 메소드입니다.