def predict(self, X): statuses = ['BAIXADA', 'NULA', 'SUSPENSA', 'INAPTA'] self._X = X.apply(self.__compare_date, axis=1) return np.r_[self._X & X['situation'].isin(statuses)]