Find Latest Salaries — LeetCode 2668 Python Solution

Table: Salary +---------------+---------+ | Column Name | Type | +---------------+---------+ | emp_id | int | | firstname | varchar | | lastname | varchar | | salary | varchar | | department_id | varchar | +---------------+---------+ (emp_id, salary) is the primary key (combination of columns with unique values) for this table. Each row contains employees details and their yearly salaries, however, some of the records are old and contain outdated salary information.

+---------------+---------+ 
| Column Name   | Type    | 
+---------------+---------+ 
| emp_id        | int     | 
| firstname     | varchar |
| lastname      | varchar |
| salary        | varchar |
| department_id | varchar |
+---------------+---------+
(emp_id, salary) is the primary key (combination of columns with unique values) for this table.
Each row contains employees details and their yearly salaries, however, some of the records are old and contain outdated salary information.

import duckdb
import pandas as pd

# Pass input tables as keyword arguments matching the SQL table names.
def solution(**tables) -> pd.DataFrame:
    con = duckdb.connect()
    for name, df in tables.items():
        con.register(name, df)
    return con.execute("""SELECT
    emp_id,
    firstname,
    lastname,
    MAX(salary) AS salary,
    department_id
FROM Salary
GROUP BY emp_id
ORDER BY emp_id;""").df()