Source code for feature_engine.feature_selection

from sklearn.base import BaseEstimator, TransformerMixin
from sklearn.utils.validation import check_is_fitted
from feature_engine.dataframe_checks import _is_dataframe, _check_input_matches_training_df
from feature_engine.variable_manipulation import _define_variables

[docs]class DropFeatures(BaseEstimator, TransformerMixin): """ DropFeatures() drops the list of variable(s) indicated by the user from the original dataframe and returns the remaining variables. Parameters ---------- features_to_drop : str or list, default=None Variable(s) to be dropped from the dataframe """ def __init__(self, features_to_drop=None): self.features_to_drop = _define_variables(features_to_drop) if len(self.features_to_drop) == 0: raise ValueError('List of features to drop cannot be empty. Please pass at least 1 variable to drop')
[docs] def fit(self, X, y=None): """ Verifies that the input X is a pandas dataframe Parameters ---------- X: pandas dataframe of shape = [n_samples, n_features] The input dataframe y: None y is not needed for this transformer. You can pass y or None. """ # check input dataframe X = _is_dataframe(X) # check for non existent columns non_existent = [x for x in self.features_to_drop if x not in X.columns] if non_existent: raise KeyError( f"Columns '{', '.join(non_existent)}' not present in the input dataframe, " f"please check the columns and enter a new list of features to drop" ) # check that user does not drop all columns returning empty dataframe if len(self.features_to_drop) == len(X.columns): raise ValueError("The resulting dataframe will have no columns after dropping all existing variables") # add input shape self.input_shape_ = X.shape return self
[docs] def transform(self, X): """ Drops the variable or list of variables indicated by the user from the original dataframe and returns a new dataframe with the remaining subset of variables. Parameters ---------- X: pandas dataframe The input dataframe from which features will be dropped Returns ------- X_transformed: pandas dataframe of shape = [n_samples, n_features - len(features_to_drop)] The transformed dataframe with the remaining subset of variables. """ # check if fit is called prior check_is_fitted(self) # check input dataframe X = _is_dataframe(X) # check for input consistency _check_input_matches_training_df(X, self.input_shape_[1]) X = X.drop(columns=self.features_to_drop) return X