unformat option improved to unformat headers aswell

raul-martin-dev · Dec 18, 2023 · 34b4dec · 34b4dec
1 parent c858d5c
commit 34b4dec
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/Driveup/drive.py b/Driveup/drive.py
@@ -158,6 +158,7 @@ def df_update(self,df:Union[pd.DataFrame, List[pd.DataFrame]],id:str,sheet_name:
             if unformat == True:
                 df = df.fillna('NULL')
                 df = df.astype(str)
+                df.columns = df.columns.astype(str)
 
             if sheet_name == None:
                 sheet_name = sheets[0].get("properties", {}).get("title", "Sheet1")
@@ -300,5 +301,7 @@ def df_download(self,id:str,sheet_name:str=None,unformat:bool = False):
         if unformat == True:
             df = df.fillna('NULL')
             df = df.astype(str)
+            df.columns = df.columns.astype(str)
+
 
         return df