Pārlūkot izejas kodu

Add method to flattendata

ogert 4 gadi atpakaļ
vecāks
revīzija
b994825cf6
1 mainītis faili ar 10 papildinājumiem un 1 dzēšanām
  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   
+