The Winner University — LeetCode 2072 Python Solution

Table: NewYork +-------------+------+ | Column Name | Type | +-------------+------+ | student_id | int | | score | int | +-------------+------+ In SQL, student_id is the primary key for this table. Each row contains information about the score of one student from New York University in an exam.

+-------------+------+
| Column Name | Type |
+-------------+------+
| student_id  | int  |
| score       | int  |
+-------------+------+
In SQL, student_id is the primary key for this table.
Each row contains information about the score of one student from New York University in an exam.

import duckdb
import pandas as pd

def solution(new_york: pd.DataFrame, california: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("NewYork", new_york)
    con.register("California", california)
    return con.execute("""SELECT
    CASE
        WHEN n1.cnt > n2.cnt THEN 'New York University'
        WHEN n1.cnt < n2.cnt THEN 'California University'
        ELSE 'No Winner'
    END AS winner
FROM
    (SELECT COUNT(1) AS cnt FROM NewYork WHERE score >= 90) AS n1,
    (SELECT COUNT(1) AS cnt FROM California WHERE score >= 90) AS n2;""").df()