remove spaces from data frame

rhiever · Jul 28, 2017 · 6b951dc · 6b951dc
1 parent 5aa9215
commit 6b951dc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/datacleaner/datacleaner.py b/datacleaner/datacleaner.py
@@ -94,7 +94,7 @@ def autoclean(input_dataframe, drop_nans=False, copy=False, encoder=None,
                 column_encoder = LabelEncoder().fit(input_dataframe[column].values)
 
             input_dataframe[column] = column_encoder.transform(input_dataframe[column].values)
-
+    input_dataframe.columns =  [c.strip() for c in input_dataframe.columns]
     return input_dataframe
 
 def autoclean_cv(training_dataframe, testing_dataframe, drop_nans=False, copy=False,
@@ -174,8 +174,7 @@ def autoclean_cv(training_dataframe, testing_dataframe, drop_nans=False, copy=Fa
             training_dataframe[column] = column_encoder.transform(training_dataframe[column].values)
             testing_dataframe[column] = column_encoder.transform(testing_dataframe[column].values)
 
-    training_dataframe.columns =  [c.strip() for c in training_dataframe.columns]
-    testing_dataframe.columns  =  [c.strip() for c in testing_dataframe.columns]
+
 
     return training_dataframe, testing_dataframe