python绘制散点图：二分类样本

在做机器学习数据集的探索时，需要绘制二变量的散点图，散点图要能够区分正负样本。
如何用python绘制散点图呢？思路其实不复杂：
1.绘制正样本，使用“+”图标
2.在同一张图上绘制负样本，使用“o”图标
3.展示正负样本

import numpy as np
import matplotlib.pyplot as plt

# # 载入数据
# data = np.loadtxt('./ex1data1.txt', delimiter=',')

#区分正负样本
pos_data=data[data[:,2]==1]
neg_data=data[data[:,2]!=1]

# 绘图
# 绘制正样本
plt.scatter(pos_data[:,0],pos_data[:,1],marker='+')
# 绘制负样本
plt.scatter(neg_data[:,0],neg_data[:,1],marker='o')
plt.show()

绘制的图形如下：