使用不同的标记python绘制类

Question

我有特征 x1 和 x2 以及类 y 的数据集列，其值为 0 或 1。我想在散点图中绘制 x1 和 x2，使得值 y == 1 将显示为“+”，值 y == 0将显示为“o”。

x1 = np.array(100)
x2 = np.array(100)
#y = array of length 100 either with value 1 or 0
plt.scatter(x1, x2, y=1, marker='+')
plt.scatter(x1, x2, y=0, marker='o')
plt.show()

有什么建议么？

Answer 1

您可以使用or的条件对x1和数组进行索引：x2y==0y==1

plt.scatter(x1[y==1], x2[y==1], marker='+')
plt.scatter(x1[y==0], x2[y==0], marker='o')