import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from sklearn.neighbors import KNeighborsClassifier
knn = KNeighborsClassifier(n_neighbors = 3)
knn.fit(X_train,Y_train)
# Visualising the Test set results
from matplotlib.colors import ListedColormap
X_set, y_set = X_test, Y_test
X1, X2 = np.meshgrid(np.arange(start = X_set[:, 0].min() - 1, stop = X_set[:, 0].max() +
1, step = 0.01),np.arange(start = X_set[:, 1].min() - 1, stop = X_set[:, 1].max() + 1,
step = 0.01))
plt.sactter(X1, X2, knn.predict(np.array([X1.ravel(),X2.ravel()]).T).reshape(X1.shape),
alpha = 0.75, cmap = ListedColormap(('red', 'green')))
plt.xlim(X1.min(), X1.max())
plt.ylim(X2.min(), X2.max())
for i, j in enumerate(np.unique(y_set)):
plt.scatter(X_set[y_set == j, 0], X_set[y_set == j, 1],c = ListedColormap(('red',
'green'))(i), label = j)
plt.title('Classifier (Test set)')
plt.xlabel('Age')
plt.ylabel('Estimated Salary')
plt.legend()
plt.show()
错误:
File "C:\Users\shaar\.spyder-py3\MLPractice\KNN.py", line 55, in <module>
plt.scatter(X_set[y_set == j, 0], X_set[y_set == j, 1])
IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed
发布于 2021-12-11 07:20:15
这通常发生在您尝试输入或尝试使用numpy数组的其他维度时,而它仅为1D。更清楚的是,如果您有一个numpy数组,如
a = [1,2,3,4]
和以后,如果您尝试使用它的值,比如使用(1,2)
,那么如果您试图找到一个2D numpy数组的第一行和第二列,它就会用到它。因此,在访问numpy数组时不要使用逗号。希望我很清楚,如果不考虑检查arrays.asp
发布于 2022-06-01 09:06:52
如果numpy数组有两个维度,但是其中一个维度只有一个块,那么numpy数组做什么就有点困惑了。在您的代码片段中,我们无法看到您在y_set
中填充了哪些内容。
X_set, y_set = X_test, Y_test
但我认为,如果您查看y_set
和y_set.shape
的维数,您将得到
(150,1)
(我假设有150个数据集)。Python将为每个形状生成一个索引。若要分隔想要的维度,可以将不需要的维度设置为零:
y_set_one_dimension = y_set[:,0]
print(y_set_one_dimension.shape)
就像在如何访问NumPy多维数组的ith列?中所描述的那样
产出如下:
(150,)
现在,散点图将获得所需的2个独立的二维,并将工作。
注释:
如果y_set
是一个dataframe,那么您必须首先将它转换为一个numpy数组:
yArray = numpy.array(y_set)
发布于 2022-07-23 04:30:39
X_set, y_set = X_test, Y_test.ravel()
https://stackoverflow.com/questions/70313265
复制相似问题