Project Employees I — LeetCode 1075 Python Solution

Table: Project +-------------+---------+ | Column Name | Type | +-------------+---------+ | project_id | int | | employee_id | int | +-------------+---------+ (project_id, employee_id) is the primary key of this table. employee_id is a foreign key to Employee table.

+-------------+---------+
| Column Name | Type    |
+-------------+---------+
| project_id  | int     |
| employee_id | int     |
+-------------+---------+
(project_id, employee_id) is the primary key of this table.
employee_id is a foreign key to Employee table.
Each row of this table indicates that the employee with employee_id is working on the project with project_id.

import duckdb
import pandas as pd

def solution(project: pd.DataFrame, employee: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Project", project)
    con.register("Employee", employee)
    return con.execute("""SELECT project_id, ROUND(AVG(experience_years), 2) AS average_years
FROM
    Project
    JOIN Employee USING (employee_id)
GROUP BY 1;""").df()