소스 검색

Add functionality to ignore certain labels from being flattend

ogert 4 년 전
부모
커밋
107eb98a52
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      cdplib/FlattenData.py

+ 1 - 1
cdplib/FlattenData.py

@@ -175,7 +175,7 @@ class FlattenData():
 
         return dictionary
 
-    def flatten_if_not_flat(self, data: pd.DataFrame):
+    def flatten_if_not_flat(self, data: pd.DataFrame, labels_to_ignore: list = None):
 
         for data_type in data.dtypes:
                 if data_type == object: