Python: как использовать функцию sklearn с фреймами данных?

Question

У меня есть датафрейм X, который содержит 40 особенности 170 городов.Чем у меня есть целевая переменная y в виде pandas.core.series.Series

X.shape
(170, 40)

type(y)
pandas.core.series.Series

y.shape
(170,)

Я хочу попробовать применить алгоритм SelectKBest из sklearn

from sklearn.feature_selection import SelectKBest
from sklearn.feature_selection import chi2

X_new = SelectKBest(chi2, k=2).fit_transform(X, y)

, но яполучил следующую ошибку

ValueError: Неизвестный тип метки: (массив ([

Answer 1

Согласно документации Scikit-learn , y должен быть массивом numpy.Попробуйте преобразовать эту переменную в numy-like arr, используя pandas.Series.values ​​:

y = y.values