绘制散点图:
import matplotlib.pyplot as plt
plt.figure(dpi = 120) #修改图像大小,必须放在scatter之前,才会有效果
plt.scatter(score1,score2)
plt.show()
分类的散点图:
import matplotlib.pyplot as plt
#找到正例子和反例子
pos = data[data[2]==1]
neg = data[data[2]==0]
plt.figure(dpi = 120) #设置图像大小
plt.plot(pos[0],pos[1],'r+',label='admitted')#用pos的第一列数据和pos的第二列数据来绘图,用红色加号来标记
plt.plot(neg[0],neg[1],'bo',label='not admitted')#用蓝色圆圈来标记
plt.legend(loc='upper right')
plt.xlabel('score1')
plt.ylabel('score2')
plt.title('Figure1:Scatter plot of training data')
plt.show()
散点图的颜色设置:
例如:c='y'