1757. Recyclable and Low Fat Products : leetcode

import pandas as pd

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

    return results[['product_id']]

data = [['0', 'Y', 'N'], ['1', 'Y', 'Y'], ['2', 'N', 'Y'], ['3', 'Y', 'Y'], ['4', 'N', 'N']]
products = pd.DataFrame(data, columns=['product_id', 'low_fats', 'recyclable']).astype({'product_id':'int64', 'low_fats':'category', 'recyclable':'category'})

result = find_products(products)
print(result)