Find Cutoff Score for Each School — LeetCode 1988 Python Solution

Table: Schools +-------------+------+ | Column Name | Type | +-------------+------+ | school_id | int | | capacity | int | +-------------+------+ school_id is the column with unique values for this table. This table contains information about the capacity of some schools.

+-------------+------+
| Column Name | Type |
+-------------+------+
| school_id   | int  |
| capacity    | int  |
+-------------+------+
school_id is the column with unique values for this table.
This table contains information about the capacity of some schools. The capacity is the maximum number of students the school can accept.

import duckdb
import pandas as pd

def solution(schools: pd.DataFrame, exam: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Schools", schools)
    con.register("Exam", exam)
    return con.execute("""SELECT school_id, MIN(IFNULL(score, -1)) AS score
FROM
    Schools AS s
    LEFT JOIN Exam AS e ON s.capacity >= e.student_count
GROUP BY school_id;""").df()