Order Two Columns Independently — LeetCode 2159 Python Solution

Table: Data +-------------+------+ | Column Name | Type | +-------------+------+ | first_col | int | | second_col | int | +-------------+------+ This table may contain duplicate rows. Write a solution to independently: order first_col in ascending order.

+-------------+------+
| Column Name | Type |
+-------------+------+
| first_col   | int  |
| second_col  | int  |
+-------------+------+
This table may contain duplicate rows.

import duckdb
import pandas as pd

def solution(data: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Data", data)
    return con.execute("""WITH
    S AS (
        SELECT
            first_col,
            ROW_NUMBER() OVER (ORDER BY first_col) AS rk
        FROM Data
    ),
    T AS (
        SELECT
            second_col,
            ROW_NUMBER() OVER (ORDER BY second_col DESC) AS rk
        FROM Data
    )
SELECT first_col, second_col
FROM
    S
    JOIN T USING (rk);""").df()