[LeetCode - Pandas] (Easy) 1757. Recyclable and Low Fat Products

 

Table: Products

+-------------+---------+
| Column Name | Type    |
+-------------+---------+
| product_id  | int     |
| low_fats    | enum    |
| recyclable  | enum    |
+-------------+---------+

product_id는 이 테이블의 기본 키(고유 값이 있는 열)입니다.
low_fats는 ('Y', 'N') 유형의 ENUM(카테고리)으로, 'Y'는 이 제품이 저지방임을 의미하고, 'N'은 그렇지 않음을 의미합니다.
recyclable은 ('Y', 'N') 유형의 ENUM(카테고리)으로, 'Y'는 이 제품이 재활용 가능함을 의미하고, 'N'은 그렇지 않음을 의미합니다.

 

문제

저지방이면서 재활용 가능한 제품들의 ID를 찾는 솔루션을 작성하세요.

결과 테이블은 아무 순서로 반환할 수 있습니다.

결과 형식은 아래 예시와 같습니다.

Example 1:

Input: 
Products table:
+-------------+----------+------------+
| product_id  | low_fats | recyclable |
+-------------+----------+------------+
| 0           | Y        | N          |
| 1           | Y        | Y          |
| 2           | N        | Y          |
| 3           | Y        | Y          |
| 4           | N        | N          |
+-------------+----------+------------+
Output: 
+-------------+
| product_id  |
+-------------+
| 1           |
| 3           |
+-------------+

설명 : 제품 1과 3만이 저지방이면서 재활용 가능합니다.

 

✏️ 풀이

import pandas as pd

def find_products(products: pd.DataFrame) -> pd.DataFrame:
    return products[(products['low_fats'] == 'Y') & (products['recyclable'] == 'Y')][['product_id']]

이제 기본적인 문법은 어느정도 익숙해져서 SQL에서 where과 절과 같은 

조건 조회는 이제 쉽게 할 수 있는 것 같다