报错TypeError: '(slice(None, 1, None), 1)' is an invalid key

Question

程序如下

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from sklearn.datasets import make_blobs
from sklearn.svm import SVC
dataset = pd.read_csv('zhendaorushuju.csv', encoding='gb18030')
Y = dataset[:+1,1]
X = dataset[:+1,:+1]
model=SVC(C=0.1,kernel='poly',degree=10)
model.fit(X,Y)
x_min,x_max=X[:,0].min()-1,X[:,0].max()+1
y_min,y_max=X[:,1].min()-1,X[:,1].max()+1
xx,yy=np.meshgrid(np.arange(x_min,x_max,0.02), np. arange(y_min,y_max,0.02))
z=model.predict(np.c_[xx.ravel(),yy.ravel()])
z=z.reshape(xx.shape)
plt.pcolormesh(xx,yy,z,cmap=plt.cm.Pastel1)
plt.scatter(X[:,0],X[:,1],s=80,c=Y,cmap=plt.cm.spring,edgecolors='k')
plt.xlim(xx.min(),xx.max())
plt.ylim(yy.min(),yy.max())
plt.title("Classify")
plt.scatter(0,2,marker='*',c='red',s=200)
plt.show()
res = model.predict([[0,2]])
print('Classification flag: '+str(res))
print('Model score:{:.2f}' .format(model.score(X,Y)))

Answer 1

你得程序在下面两行中有问题：

Y = dataset[:+1,1]
X = dataset[:+1,:+1]

具体来说，pandas中的数据结构不支持这种索引方式。你可以使用iloc来进行行或者列的提取，例如

# 提取第一例数据
Y = dataset.iloc[:, 1].values
# 提取前两列数据
X = dataset.iloc[:, :2].values