# Sample Python Polars code to compare with the `perf_test.rs` example. The
# following is based on code from Alexander Beedie in the following GitHub
# thread: https://github.com/pola-rs/polars/issues/5568#issuecomment-1526316286
from codetiming import Timer
from datetime import date
import polars as pl
# Quickly spin-up a 1,000,000 element DataFrame.
df = pl.DataFrame(
{
"Int": range(250_000),
"Float": 123.456789,
"Date": date.today(),
"String": "Test"
}
)
# Export to Excel from polars.
with Timer():
df.write_excel("dataframe_pl.xlsx")
# Export to Excel from pandas.
pf = df.to_pandas()
with Timer():
pf.to_excel("dataframe_pd.xlsx", index=False)