0
定制变压器我下面的sklearn_pandas穿行在sklearn_pandas README on github发现,我试图修改DateEncoder()定制变压器为例做2个额外的东西:其将日期,新列
- 转换字符串类型的列以日期格式作为参数时的日期时间
- 当吐出新列时附加原始列名称。例如:如果输入列:Date1则输出:Date1_year,Date1_month,Date_1日。
这里是我的尝试(与sklearn管道的一个相当基本的了解):
import pandas as pd
import numpy as np
from sklearn.base import TransformerMixin, BaseEstimator
from sklearn_pandas import DataFrameMapper
class DateEncoder(TransformerMixin):
'''
Specify date format using python strftime formats
'''
def __init__(self, date_format='%Y-%m-%d'):
self.date_format = date_format
def fit(self, X, y=None):
self.dt = pd.to_datetime(X, format=self.date_format)
return self
def transform(self, X):
dt = X.dt
return pd.concat([dt.year, dt.month, dt.day], axis=1)
data = pd.DataFrame({'dates1': ['2001-12-20','2002-10-21','2003-08-22','2004-08-23',
'2004-07-20','2007-12-21','2006-12-22','2003-04-23'],
'dates2' : ['2012-12-20','2009-10-21','2016-08-22','2017-08-23',
'2014-07-20','2011-12-21','2014-12-22','2015-04-23']})
DATE_COLS = ['dates1', 'dates2']
Mapper = DataFrameMapper([(i, DateEncoder(date_format='%Y-%m-%d')) for i in DATE_COLS], input_df=True, df_out=True)
test = Mapper.fit_transform(data)
但在运行时,我收到以下错误:
AttributeError: Can only use .dt accessor with datetimelike values
为什么我收到这个错误和如何解决它? 任何帮助与上面提到的原始列重命名列名(Date1_year,Date1_month,Date_1天)将不胜感激!
您在'fit'中将'X'转换为'self.dt'处的日期时间,但'transform()'不能与'self.dt'一起使用。 'X.dt'因为'X'不是datetime类型而失败。 –