From 3bb3fd33e9742f9c41269d6691a755459c014296 Mon Sep 17 00:00:00 2001
From: unknown <Yongkang>
Date: Sun, 22 Sep 2024 13:13:06 +0800
Subject: [PATCH] perf: Improved to add names to data
---
geochemistrypi/data_mining/utils/base.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/geochemistrypi/data_mining/utils/base.py b/geochemistrypi/data_mining/utils/base.py
index 84b3965..202f049 100644
--- a/geochemistrypi/data_mining/utils/base.py
+++ b/geochemistrypi/data_mining/utils/base.py
@@ -216,7 +216,8 @@ def save_data(df: pd.DataFrame, name_column: str, df_name: str, local_data_path:
Whether to write the index.
"""
if name_column is not None and len(df) == len(name_column):
- name_column = name_column.loc[df.index].reset_index(drop=True)
+ if not df.index.empty and len(name_column.index) == len(df.index) and set(df.index) == set(name_column.index):
+ name_column = name_column.reindex(df.index)
df.reset_index(drop=True, inplace=True)
name_column.reset_index(drop=True, inplace=True)
df = pd.concat([name_column, df], axis=1)