Parcourir la source

Add method to flattendata

ogert il y a 4 ans
Parent
commit
b994825cf6
1 fichiers modifiés avec 10 ajouts et 1 suppressions
  1. 10 1
      cdplib/FlattenData.py

+ 10 - 1
cdplib/FlattenData.py

@@ -170,4 +170,13 @@ class FlattenData():
         for key in to_append:
             dictionary[key] = to_append[key]
 
-        return dictionary
+        return dictionary
+
+    def flatten_if_not_flat(self, data: pd.DataFrame):
+
+        for data_type in data.dtypes:
+                if data_type == object:
+                    return self.flatten(data) 
+
+        return data   
+