`ColumnTransformer.fit_transform()`的结果只包含后面transfromer的结果

Question

ColumnTransformer 中有 2 个转换器。但是ColumnTransformer.fit_transform()的结果只包含后面transfromer的结果：

pos_time
array([[1.24100000e+03, 6.27000000e+02, 1.56279701e+09],
       [1.27100000e+03, 6.90000000e+02, 1.56279701e+09],
       [1.30200000e+03, 7.49000000e+02, 1.56279701e+09],
       ...,
       [1.81600000e+03, 8.60000000e+01, 1.56279703e+09],
       [1.81600000e+03, 8.60000000e+01, 1.56279703e+09],
       [1.81600000e+03, 8.60000000e+01, 1.56279703e+09]])

我认为这是因为 TimeTransformer 中的这一行 X=X.drop('Time',axis=1)。如果我注释掉该行，ColumnTransformer.fit_transform() 的结果将是：

pos_time
array([[1241.0, 627.0, Timestamp('2019-07-10 22:16:46.036385'),
        1562797006.036385],
       [1271.0, 690.0, Timestamp('2019-07-10 22:16:46.052012'),
        1562797006.052012],
       [1302.0, 749.0, Timestamp('2019-07-10 22:16:46.067638'),
        1562797006.067638],
       ...,
       [1816.0, 86.0, Timestamp('2019-07-10 22:17:08.327709'),
        1562797028.327709],
       [1816.0, 86.0, Timestamp('2019-07-10 22:17:08.496155'),
        1562797028.496155],
       [1816.0, 86.0, Timestamp('2019-07-10 22:17:08.585392'),
        1562797028.585392]], dtype=object)

但我不想获取第 3 列。我想知道为什么会发生这种情况以及如何解决它。谢谢！

这是我的代码。这个PositionTransformer把一个字符串转换成x,y坐标：

def position(string):
    x_,y_=string.lstrip('(').rstrip(')').split(',')
    x_,y_=float(x_),float(y_) 
    return x_,y_
class PositionTransformer(BaseEstimator,TransformerMixin):
    def __init__(self):
        pass
    def fit(self,X,y=None):
        return self
    def transform(self,X,y=None):
        print(type(X))
        print(X.head())
        print(X.shape)
        X=X['Position']
        xy=X.apply(position)
        x=xy.apply(lambda x:x[0])
        y=xy.apply(lambda x:x[1])
        xy=np.c_[x.values,y.values]
        return xy

这个TimeTransformer将字符串转换成时间戳：

class TimeTransformer(BaseEstimator,TransformerMixin):
    def __init__(self):
        pass
    def fit(self,X,y=None):
        return self
    def transform(self,X,y=None):
        X['Time']=pd.to_datetime(X['Time'])
        X['UnixTime']=X.apply(lambda row:row['Time'].timestamp(), axis=1)
        X=X.drop('Time',axis=1)
        return X

这是ColumnTransformer:

time=['Time']
pos=['Position']
time_pos_transformer=ColumnTransformer([
    ('pos_transformer',PositionTransformer(),pos),
    ('time_transformer',TimeTransformer(),time),
])
pos_time=time_pos_transformer.fit_transform(df)

Answer 1

对不起，ColumnTransformer.fit_transform()的结果是正确的。我出错的原因是当我打印pos_time时，控制台只输出了它的一部分。

`ColumnTransformer.fit_transform()`的结果只包含后面transfromer的结果

The result of `ColumnTransformer.fit_transform()` only contains the later transfromer's result

python

scikit-learn

sklearn-pandas