拆分数据时出现问题： 键错误： "None of [Int64Index([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13], dtype='int64')]

Question

我试图在一些数据上执行训练测试分割，wine。数据，但当初始化x和y时:

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from sklearn.neural_network import MLPClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
from sklearn.preprocessing import StandardScaler
from sklearn.model_selection import cross_val_score
wine =  pd.read_csv("https://archive.ics.uci.edu/ml/machine-learning-databases/wine/wine.data")
print(wine.shape)
wine.head()
X = wine[np.arange(1,14)]
y = wine[0]

当我得到错误消息时，此段下面的其余代码将不会运行:

KeyError: "None of [Int64Index([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13], dtype='int64')] are in the [columns]"

我试图通过改变X值的范围或改变np来解决这个问题。安排函数，但都不能解决问题。

任何帮助或建议将不胜感激，谢谢!

Answer 1

您忘记将header=None添加到数据框架构造函数中。您正在下载的csv文件没有标题行。因此，如果您不指定header=None，则第一行数据将用作标头。

与试

wine =  pd.read_csv(
"https://archive.ics.uci.edu/ml/machine-learning-databases/wine/wine.data",
header=None
)

Answer 2

您是否尝试按列的位置选择列?如果是这样，试试:

X = wine.iloc[:,np.arange(1,14)]
y = wine.iloc[:, 0]