NPV Queries — LeetCode 1421 Python Solution

Table: NPV +---------------+---------+ | Column Name | Type | +---------------+---------+ | id | int | | year | int | | npv | int | +---------------+---------+ (id, year) is the primary key (combination of columns with unique values) of this table. The table has information about the id and the year of each inventory and the corresponding net present value.

+---------------+---------+
| Column Name   | Type    |
+---------------+---------+
| id            | int     |
| year          | int     |
| npv           | int     |
+---------------+---------+
(id, year) is the primary key (combination of columns with unique values) of this table.
The table has information about the id and the year of each inventory and the corresponding net present value.

import duckdb
import pandas as pd

def solution(npv: pd.DataFrame, queries: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("NPV", npv)
    con.register("Queries", queries)
    return con.execute("""SELECT q.*, IFNULL(npv, 0) AS npv
FROM
    Queries AS q
    LEFT JOIN NPV AS n USING (id, year);""").df()