利用mglearn绘图报错：TypeError: ‘(slice(None, None, None), 0)‘ is an invalid key

想利用mglearn可视化有俩个特征的数据时，出现了如上的报错。data = pd.read_csv('ex2data1.txt', names=['score1', 'score2', 'Admission'])m, n = data.shapex = data[['score1', 'score2']]y = data[['Admission']]mglearn.discrete_scatte

Jay_Mapp

7227人浏览 · 2022-04-01 18:55:02

Jay_Mapp · 2022-04-01 18:55:02 发布

想利用mglearn可视化有俩个特征的数据时，出现了文章标题所示的错误。

data = pd.read_csv('ex2data1.txt', names=['score1', 'score2', 'Admission'])
m, n = data.shape
x = data[['score1', 'score2']]
y = data[['Admission']]

mglearn.discrete_scatter(x[:, 0], x[:, 1], y)
plt.show()

这是因为没有将数据转化为数组的格式，所以首先将数据转化为数组的格式。使用np.array

data = pd.read_csv('ex2data1.txt', names=['score1', 'score2', 'Admission'])
m, n = data.shape
x = data[['score1', 'score2']]
y = data['Admission']

x = np.array(x.values)
y = np.array(y.values)

mglearn.discrete_scatter(x[:, 0], x[:, 1], y)
plt.legend(['score1', 'score2'])
plt.show()

这样，问题解决，数据可视化更加方便了。