Я предполагаю, что вы ищете способы доступа к результату преобразователя, который выдает массив numpy.
ColumnTransfomer имеет атрибут с именем transformers_
: `
Из документации:
transformers_ : list
The collection of fitted transformers as tuples of
(name, fitted_transformer, column). `fitted_transformer` can be an
estimator, 'drop', or 'passthrough'. In case there were no columns
selected, this will be the unfitted transformer.
If there are remaining columns, the final element is a tuple of the
form:
('remainder', transformer, remaining_columns) corresponding to the
``remainder`` parameter. If there are remaining columns, then
``len(transformers_)==len(transformers)+1``, otherwise
``len(transformers_)==len(transformers)``.
Так что, к сожалению, предоставляется только информация о самом трансформаторе и столбце, к которому он был применен, но не о местоположении результирующие данные, за исключением следующих:
примечания: порядок столбцов в преобразованной матрице объектов соответствует порядку того, как столбцы указаны в списке transformers
.
Итак, мы знаем, что порядок выходных столбцов совпадает с порядком, в котором столбцы указаны в списке преобразователей. Кроме того, мы также знаем для наших шагов преобразования, сколько столбцов они дают, поскольку StandardScaler () выдает то же количество столбцов, что и исходные данные, а OneHotEncoder () - количество столбцов, равное количеству категорий.
import numpy as np
import pandas as pd
from sklearn.preprocessing import OneHotEncoder,StandardScaler
from sklearn.compose import ColumnTransformer, make_column_transformer
df = pd.DataFrame({'brand' : ['aaaa', 'asdfasdf', 'sadfds', 'NaN'],
'category' : ['asdf','asfa','asdfas','asd'],
'num1' : [1, 1, 0, 0] ,
'target' : [0.2,0.11,1.34,1.123]})
train_continuous_cols = df.select_dtypes(include=["int64","float64"]).columns.tolist()
train_categorical_cols = df.select_dtypes(include=["object"]).columns.tolist()
# get n_categories for categorical features
n_categories = [df[x].nunique() for x in train_categorical_cols]
preprocess = make_column_transformer(
(StandardScaler(),train_continuous_cols),
(OneHotEncoder(), train_categorical_cols)
)
preprocessed_df = preprocess.fit_transform(df)
# the scaler yield 1 column each
indexes_scaler = list(range(0,len(train_continuous_cols)))
# the encoder yields a number of columns equal to the number of categories in the data
cum_index_encoder = [0] + list(np.cumsum(n_categories))
# the encoder indexes come after the scaler indexes
start_index_encoder = indexes_scaler[-1]+1
indexes_encoder = [x + start_index_encoder for x in cum_index_encoder]
# get both lower and uper bound of index
index_pairs= zip (indexes_encoder[:-1],indexes_encoder[1:])
Это приводит к следующему выводу:
print ('Transformed {} continious cols resulting in a df with shape:'.format(len(train_continuous_cols)))
print (preprocessed_df[: , indexes_scaler].shape)
Преобразованные 2 непрерывных столбца, приводящие к df с формой: (4, 2)
for column, (start_id, end_id) in zip (train_categorical_cols,index_pairs):
print('Transformed column {} resulted in a df with shape:'.format(column))
print(preprocessed_df[:, start_id:end_id].shape)
Марка преобразованной колонны привела к df с формой: (4, 4)
Категория трансформированной колонки привела к df с формой: (4, 4)