python - 具有决策边界的两个特征的多标签分类的散点图

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
... ... ... ...         .....

标签： pythonmatplotlibscatter-plotmultilabel-classificationsupervised-learning

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