From 12a37191a4d240badb93962875bcbe42644f62ba Mon Sep 17 00:00:00 2001 From: Tomasini Luca Date: Thu, 19 Dec 2024 15:16:00 +0100 Subject: [PATCH] Modify table_to_gpkg --- src/general_function.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/general_function.py b/src/general_function.py index f215c38..c308feb 100644 --- a/src/general_function.py +++ b/src/general_function.py @@ -195,14 +195,16 @@ def convert_list_to_string(list_data: list) -> str: def table_to_gpkg(table: pl.DataFrame, gpkg_file_name: str, layer_name: str): """ - Save a Polars DataFrame as a GeoPackage file. + Save a Polars DataFrame as a GeoPackage file. As GeoPackage does not support list columns, + the list columns are joined into a single string separated with a comma. Args: table (pl.DataFrame): The Polars DataFrame. gpkg_file_name (str): The GeoPackage file name. layer_name (str): The layer name. """ - list_columns: list[str] = [name for name, col_type in dict(table.schema).items() if type(col_type) == pl.List] + list_columns: list[str] = [ + name for name, col_type in dict(table.schema).items() if type(col_type) == pl.List] table_pd: pd.DataFrame = table.with_columns( c(list_columns).list.join(", ") ).to_pandas()