Почему fit_transform SimpleImputer не работает для фрейма данных в Google Colab?

Question

imp = SimpleImputer(missing_values=np.nan, strategy='most_frequent')
weather_test = imp.fit_transform(weather_test)

Приведенный выше код выдает ошибку в google colab, когда weather_test представляет собой pandas фрейм данных. Но когда я изменяю weather_test на массив numpy, он работает.

imp = SimpleImputer(missing_values=np.nan, strategy='most_frequent', verbose=0)
weather_test = imp.fit_transform(np.array(weather_test))

Answer 1

Методы API Scikit-learn обычно предполагают, что входными данными будет массив numpy, а не pandas фрейм данных. Пример использования этой функции с фреймом данных см. В Как использовать sklearn fit_transform с pandas и возвращать фрейм данных вместо numpy массива?