Передача столбцов индекса в sklearn.compose.ColumnTransformer - PullRequest
0 голосов
/ 17 ноября 2018

Я создал класс DateTransformer, который извлекает такие функции, как год, месяц и т. Д., Из pd.TimeStamp, когда «timestamp» является столбцом

from sklearn.pipeline import Pipeline
from sklearn.compose import ColumnTransformer

ct = ColumnTransformer([("date", DateTransformer(),['timestamp'])])
ct.fit_transform(data)

, все работает нормально. Однако в моем случаеотметка времени на самом деле является уровнем индекса, поэтому вместо доступа к коллекции столбцов мне нужно, чтобы ColumnTransformer выполнил следующее

data.index.get_level_values('timestamp')

Я могу сделать pd.reset_index(), но мне было интересно, есть ли другой способ?

...