aoc24-day-1.py

import polars as pl
import pandas as pd

# Read input data as a data frame with two columns
input = pl.from_pandas(
  pd.read_table("day-1/input", sep = "   ", header = None, 
                names = ["list_1", "list_2"])
)

# Sort each column separately, 
# then compute difference between each column's respective entries,
# and sum up the differences
input.with_columns(
    pl.col(nm).sort() for nm in input.columns
).select(
    (pl.col("list_1") - pl.col("list_2")).abs().sum().alias("list_distance")
)