• data analysis/PYTHON
1068. Product Sales Analysis I
oujin
2024. 2. 22. 22:17
728x90
테이블 생성코드
data = [[1, 100, 2008, 10, 5000], [2, 100, 2009, 12, 5000], [7, 200, 2011, 15, 9000]]
sales = pd.DataFrame(data, columns=['sale_id', 'product_id', 'year', 'quantity', 'price']).astype({'sale_id':'Int64', 'product_id':'Int64', 'year':'Int64', 'quantity':'Int64', 'price':'Int64'})
data = [[100, 'Nokia'], [200, 'Apple'], [300, 'Samsung']]
product = pd.DataFrame(data, columns=['product_id', 'product_name']).astype({'product_id':'Int64', 'product_name':'object'})
문제
해결코드
import pandas as pd
def sales_analysis(sales: pd.DataFrame, product: pd.DataFrame) -> pd.DataFrame:
left_merge_df = pd.merge(sales,product, on='product_id')
result_df = left_merge_df[['product_name','year','price']]
return result_df
해설
sales 테이블과 product 테이블을 product_id 라는 이름의 컬럼명을 기준으로 합쳐준다.
---> left_merge_df = pd.merge(sales,product, on='product_id')
합친 테이블인 left_merge_df 테이블에서 원하는 컬럼명만을 추출하여 새로운 데이터 프레임을 만든다
---> result_df = left_merge_df[['product_name','year','price']]
return 한다
---> return result_df
728x90