diff --git a/qlib/data/dataset/processor.py b/qlib/data/dataset/processor.py index 66db55323..308c531b9 100755 --- a/qlib/data/dataset/processor.py +++ b/qlib/data/dataset/processor.py @@ -112,12 +112,12 @@ class Fillna(Processor): """Process infinity """ def __call__(self, df): - def fill_na(df): - for col in df.columns: - # FIXME: Such behavior is very weird - df[col] = df[col].fillna(0) + def fill_na(df, columns=None, fill=0): + + if columns == None: + columns = df.columns + df[columns] = df[columns].fillna(fill) - df.sort_index(inplace=True) return df return fill_na(df)