SVM做鸢尾花分类预测 KeyError: “None of [Int64Index([0, 1, 2, 3], dtype=‘int64‘)] are in the [columns]错误探索

用支持向量机做鸢尾花分类预测时
敲代码有个关于KeyError的报错，如下：
KeyError: “None of [Int64Index([0, 1, 2, 3], dtype=‘int64’)] are in the [columns]”
原始代码如下

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn import svm
from sklearn.metrics import accuracy_score
import numpy as np
data = pd.read_csv(r"G:\实验6/iris.csv")  
x, y = data[range(4)], data[4]  
y = pd.Categorical(y).codes  
x = x[[0,1