// SPDX-License-Identifier: MIT OR Apache-2.0
//
// Copyright 2022-2025, John McNamara, jmcnamara@cpan.org
//! An example of writing a Polar Rust dataframe to an Excel file. This
//! demonstrates setting formats for different columns.
use polars::prelude::*;
use polars_excel_writer::PolarsExcelWriter;
fn main() -> PolarsResult<()> {
// Create a sample dataframe for the example.
let df: DataFrame = df!(
"East" => &[1.0, 2.22, 3.333, 4.4444],
"West" => &[1.0, 2.22, 3.333, 4.4444],
)?;
// Create a new excel writer.
let mut excel_writer = PolarsExcelWriter::new();
// Set the number formats for the columns.
excel_writer.set_column_format("East", "0.00");
excel_writer.set_column_format("West", "0.0000");
// Write the dataframe to Excel.
excel_writer.write_dataframe(&df)?;
// Save the file to disk.
excel_writer.save("dataframe.xlsx")?;
Ok(())
}