diff --git a/quickda/clean_data.py b/quickda/clean_data.py index 7c47cd1fe..5abd5dd32 100644 --- a/quickda/clean_data.py +++ b/quickda/clean_data.py @@ -42,7 +42,7 @@ def fill_na_rows(data): # Retrieve the list of columns having nulls and interpolate for each column na_columns = df.columns[df.isna().any()].tolist() for column in na_columns: - df[column] = df[column].interpolate(method='pad', limit_direction='forward') + df[column] = df[column].interpolate(method='linear', limit_direction='both') return df @@ -185,4 +185,4 @@ def clean(data, method="default", columns=[], dtype="numeric", to_replace="", va except Exception as e: print(e) - return data \ No newline at end of file + return data