数据的前两列为成绩,第三列为是否录取的决定。
- 首先加载数据,将数据打在散点图上
data = np.loadtxt('ex2data1.txt', delimiter=',')
neg = data[data[:, 2] == 0]
pos = data[data[:, 2] == 1]
fig, ax = plt.subplots()
ax.scatter(pos[:, 0], pos[:, 1], c='r', marker='+', label='Admitted')
ax.scatter(neg[:, 0], neg[:, 1], c='g', marker='o', label='Not admitted')
ax.set_xlabel('Exame 1 score')
ax.set_ylabel('Exame 2 score')
plt.show()