make_column_selector

classsklearn.compose.make_column_selector(pattern=None, *, dtype_include=None, dtype_exclude=None)[source]

Создает вызываемый объект для выбора столбцов, который используется с ColumnTransformer.

make_column_selector может выбирать столбцы по типу данных или по их именам с помощью регулярного выражения. При использовании нескольких критериев выбора столбец выбирается, только если ему соответствуют все критерии.

Пример использования make_column_selector внутри ColumnTransformer для выбора столбцов по типу данных (например, dtype) см. в разделе Column Transformer with Mixed Types.

Параметры:

patternstr, default=None

Будут включены столбцы, имена которых содержат этот шаблон регулярного выражения. Если значение равно None, столбцы не выбираются на основе шаблона.

dtype_includecolumn dtype or list of column dtypes, default=None

Типы данных, которые следует включить. Подробнее см. в документации pandas.DataFrame.select_dtypes.

dtype_excludecolumn dtype or list of column dtypes, default=None

Типы данных, которые следует исключить. Подробнее см. в документации pandas.DataFrame.select_dtypes.

Возвращает:

selectorcallable

Вызываемый объект для выбора столбцов, используемый ColumnTransformer.

См. также

ColumnTransformer

Класс, позволяющий объединять результаты нескольких объектов-преобразователей, примененных к подмножествам столбцов данных, в единое пространство признаков.

Примеры

>>> from sklearn.preprocessing import StandardScaler, OneHotEncoder
>>> from sklearn.compose import make_column_transformer
>>> from sklearn.compose import make_column_selector
>>> import numpy as np
>>> import pandas as pd
>>> X = pd.DataFrame({'city': ['London', 'London', 'Paris', 'Sallisaw'],
...                   'rating': [5, 3, 4, 5]})
>>> ct = make_column_transformer(
...       (StandardScaler(),
...        make_column_selector(dtype_include=np.number)),  # rating
...       (OneHotEncoder(),
...        make_column_selector(dtype_include=[object, "string"])))  # city
>>> ct.fit_transform(X)
array([[ 0.90453403,  1.        ,  0.        ,  0.        ],
       [-1.50755672,  1.        ,  0.        ,  0.        ],
       [-0.30151134,  0.        ,  1.        ,  0.        ],
       [ 0.90453403,  0.        ,  0.        ,  1.        ]])

__call__(df)[source]

Вызываемый объект для выбора столбцов, используемый ColumnTransformer.

Параметры:

dfdataframe of shape (n_features, n_samples)

DataFrame, из которого выбираются столбцы.

Примеры из галереи

Column Transformer with Mixed Types

Categorical Feature Support in Gradient Boosting

Evaluation of outlier detection estimators