首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >IndexError :数组的索引太多了:数组是二维的,但是有3个索引是索引的。

IndexError :数组的索引太多了:数组是二维的,但是有3个索引是索引的。
EN

Stack Overflow用户
提问于 2021-12-11 07:08:36
回答 3查看 6.3K关注 0票数 -1
代码语言:javascript
运行
复制
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()

错误:

代码语言:javascript
运行
复制
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
EN

回答 3

Stack Overflow用户

发布于 2021-12-11 07:20:15

这通常发生在您尝试输入或尝试使用numpy数组的其他维度时,而它仅为1D。更清楚的是,如果您有一个numpy数组,如

a = [1,2,3,4]和以后,如果您尝试使用它的值,比如使用(1,2),那么如果您试图找到一个2D numpy数组的第一行和第二列,它就会用到它。因此,在访问numpy数组时不要使用逗号。希望我很清楚,如果不考虑检查arrays.asp

票数 0
EN

Stack Overflow用户

发布于 2022-06-01 09:06:52

如果numpy数组有两个维度,但是其中一个维度只有一个块,那么numpy数组做什么就有点困惑了。在您的代码片段中,我们无法看到您在y_set中填充了哪些内容。

代码语言:javascript
运行
复制
X_set, y_set = X_test, Y_test

但我认为,如果您查看y_sety_set.shape的维数,您将得到

代码语言:javascript
运行
复制
(150,1)

(我假设有150个数据集)。Python将为每个形状生成一个索引。若要分隔想要的维度,可以将不需要的维度设置为零:

代码语言:javascript
运行
复制
y_set_one_dimension = y_set[:,0]
print(y_set_one_dimension.shape)

就像在如何访问NumPy多维数组的ith列?中所描述的那样

产出如下:

代码语言:javascript
运行
复制
(150,)

现在,散点图将获得所需的2个独立的二维,并将工作。

注释:

如果y_set是一个dataframe,那么您必须首先将它转换为一个numpy数组:

代码语言:javascript
运行
复制
yArray = numpy.array(y_set)
票数 0
EN

Stack Overflow用户

发布于 2022-07-23 04:30:39

代码语言:javascript
运行
复制
X_set, y_set = X_test, Y_test.ravel()
票数 -1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/70313265

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档