Как я могу преобразовать из фрейма данных в массив

Question

Я пытаюсь преобразовать Dataframe в массив с использованием Python 2.7, но я получил ошибку.

Это мой код:

train = arff.loadarff(open('/Users/cristinamulas/train.arff', 'r')) 
train = pd.DataFrame([train])   
train = train[['sepal_length','sepal_width','petal_length','petal_width','CLASS_LABEL']].values 

Ошибка

File "/anaconda2/lib/python2.7/site-packages/pandas/core/frame.py", line 2726, in _getitem_array
indexer = self.loc._convert_to_indexer(key, axis=1)
File "/anaconda2/lib/python2.7/site-packages/pandas/core/indexing.py", line 1327, in _convert_to_indexer.format(mask=objarr[mask]))
KeyError: "['sepal_length' 'sepal_width' 'petal_length' 'petal_width' 'CLASS_LABEL'] not in index"

Answer 1

Попробуйте непосредственно следующий код

train.values

Удалить имена столбцов из строки кода.