Find Customers With Positive Revenue this Year — LeetCode 1821 Python Solution

Table: Customers +--------------+------+ | Column Name | Type | +--------------+------+ | customer_id | int | | year | int | | revenue | int | +--------------+------+ (customer_id, year) is the primary key (combination of columns with unique values) for this table. This table contains the customer ID and the revenue of customers in different years.

+--------------+------+
| Column Name  | Type |
+--------------+------+
| customer_id  | int  |
| year         | int  |
| revenue      | int  |
+--------------+------+
(customer_id, year) is the primary key (combination of columns with unique values) for this table.
This table contains the customer ID and the revenue of customers in different years.
Note that this revenue can be negative.

import duckdb
import pandas as pd

def solution(customers: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Customers", customers)
    return con.execute("""SELECT
    customer_id
FROM Customers
WHERE year = '2021' AND revenue > 0;""").df()