具有决策边界的双特征多标签分类散点图

Question

包含两个标签0,1的数据集，如何为其绘制散点图

    Age EstimatedSalary Purchased
0   19  19000             0
1   35  20000             0
2   26  43000             0
3   27  57000             0
4   19  76000             0 
... ... ... ...         .....
395 46  41000             1
396 51  23000             1
397 50  20000             1
... ... ... ...         .....

Answer 1

将“c”参数设置为目标变量

data.head()

X = data.iloc[:,0:2].values
y = data.iloc[:,2:3].values
plt.scatter(X[:,0:1], X[:,1:2], c=y)
plt.show