Account Balance — LeetCode 2066 Python Solution

Table: Transactions +-------------+------+ | Column Name | Type | +-------------+------+ | account_id | int | | day | date | | type | ENUM | | amount | int | +-------------+------+ (account_id, day) is the primary key (combination of columns with unique values) for this table. Each row contains information about one transaction, including the transaction type, the day it occurred on, and the amount.

+-------------+------+
| Column Name | Type |
+-------------+------+
| account_id  | int  |
| day         | date |
| type        | ENUM |
| amount      | int  |
+-------------+------+
(account_id, day) is the primary key (combination of columns with unique values) for this table.
Each row contains information about one transaction, including the transaction type, the day it occurred on, and the amount.
type is an ENUM (category) of the type ('Deposit','Withdraw')

import duckdb
import pandas as pd

def solution(transactions: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Transactions", transactions)
    return con.execute("""SELECT
    account_id,
    day,
    SUM(IF(type = 'Deposit', amount, -amount)) OVER (
        PARTITION BY account_id
        ORDER BY day
    ) AS balance
FROM Transactions
ORDER BY 1, 2;""").df()