Как мне получить доступ ко всем столбцам DataFrame панд, кроме первого?

Question

У меня возникли некоторые проблемы с sklearn.cluster.

Я подготовил данные к кластеризации и у меня есть несколько столбцов с данными типа float.

Я проверял дважды, столбцыdtype - это float64, но когда я попытался

df['cluster'] = cluster.fit_predict([df.columns[1:])

, я вижу:

ValueError: could not convert string to float: column_name_1

Last Traceback выглядит как

       ...
   -> return self.fit(X).lables
       ...
   -> X = Self._check_fit_data(X)
       ...
   -> X = check_array(X, accept_sparce='csr', dtype = [np.float, np.float32])
       ...
   -> array = np.array(array, dtype=dtype, order=order, copy=copy)

Я пытался преобразовать float в строку иназад, но это не работает.Что я должен попытаться решить эту проблему?PS Я использую Python 2.7.

Answer 1

Вы пытаетесь вписать имена столбцов.

df['cluster'] = cluster.fit_predict([df.columns[1:])

Должно быть

df['cluster'] = cluster.fit_predict(df.loc[:,1:])