Warehouse Manager — LeetCode 1571 Python Solution

Table: Warehouse +--------------+---------+ | Column Name | Type | +--------------+---------+ | name | varchar | | product_id | int | | units | int | +--------------+---------+ (name, product_id) is the primary key (combination of columns with unique values) for this table. Each row of this table contains the information of the products in each warehouse.

+--------------+---------+
| Column Name  | Type    |
+--------------+---------+
| name         | varchar |
| product_id   | int     |
| units        | int     |
+--------------+---------+
(name, product_id) is the primary key (combination of columns with unique values) for this table.
Each row of this table contains the information of the products in each warehouse.

import duckdb
import pandas as pd

def solution(warehouse: pd.DataFrame, products: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Warehouse", warehouse)
    con.register("Products", products)
    return con.execute("""SELECT
    name AS warehouse_name,
    SUM(width * length * height * units) AS volume
FROM
    Warehouse
    JOIN Products USING (product_id)
GROUP BY 1;""").df()