如何使用sklearn.base.SimpleImputer但返回pandas DataFrame来实现sklearn转换器

import pandas as pd
from sklearn.base import BaseEstimator, TransformerMixin
from sklearn.impute import SimpleImputer

class DataFrameImputer(BaseEstimator, TransformerMixin):
    def __init__(self, strategy='mean', fill_value=None):
        self.strategy = strategy
        self.fill_value = fill_value
        self.imputer = SimpleImputer(strategy=self.strategy, fill_value=self.fill_value)

    def fit(self, X, y=None):
        self.imputer.fit(X)
        return self

    def transform(self, X, y=None):
        X_transformed = self.imputer.transform(X)
        return pd.DataFrame(X_transformed, columns=X.columns)

# 创建一个示例DataFrame
data = pd.DataFrame({'A': [1, 2, np.nan, 4, 5],
                     'B': [6, np.nan, 8, 9, 10],
                     'C': [11, 12, 13, np.nan, 15]})

# 创建DataFrameImputer实例
imputer = DataFrameImputer(strategy='mean')

# 调用fit_transform方法进行数据转换
transformed_data = imputer.fit_transform(data)

# 打印转换后的DataFrame
print(transformed_data)