qurator-spk/modstool
1
0
Fork 0
mirror of https://github.com/qurator-spk/modstool.git synced 2025-08-28 19:09:53 +02:00
Code Issues Releases Wiki Activity

🐛 Fix adding index columns as columns for MultiIndexes

Browse source
This commit is contained in:
Mike Gerber 2025-08-19 19:30:01 +02:00
parent 18ad4fc854
commit 9eaac4d86d
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/mods4pandas/lib.py
View file

@ -423,8 +423,12 @@ def insert_into_db_multiple(con, table, ld: List[Dict]):
def convert_db_to_parquet(con, table, index_col, output_file):
df = pd.read_sql_query(f"SELECT * FROM {table}", con, index_col)
# Add index column as regular column, too
df[index_col] = df.index
# Add index column(s) as regular column(s), too
if isinstance(df.index, pd.MultiIndex):
for ic in index_col:
df[ic] = df.index.get_level_values(ic)
else:
df[index_col] = df.index
# Convert Python column type into Pandas type
for c in df.columns: