浏览代码

Add method to flattendata

ogert 4 年之前
父节点
当前提交
b994825cf6
共有 1 个文件被更改,包括 10 次插入1 次删除
  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   
+