先贴上源码;
import os
path='E:\\1\\kaoshi.txt'
pdData = pd.read_csv(path,header=None,names=['Exam1','Exam2','Admitted'])
pdData.head()
positive = pdData[pdData['Admitted'] == 1]
negative = pdData[pdData['Admitted'] == 0]
print(positive)
fig,ax=plt.subplots(figsize=(10,5))
ax.scatter(positive['Exam1'],positive['Exam2'],s=30,c='b',marker='o',label='Admitted')
ax.scatter(negative['Exam1'],negative['Exam2'],s=30,c='r',marker='x',label='Not Admitted')
ax.legend()
ax.set_xlabel('Exam1 Score')
ax.set_ylabel('Exam2 Score')
按视频里面编辑的源码,却发现字段‘admitted’里面的值1和0不是int型而是字符串类型的,更改
positive = pdData[pdData['Admitted'] == 1]
negative = pdData[pdData['Admitted'] == 0]为
positive = pdData[pdData['Admitted'] == '1']
negative = pdData[pdData['Admitted'] == '0']
错误解决