带有标记数据的散点图，与标签相关的标记

Question

我想用标记数据散点图数据集。我希望以不同的标记样式显示几个类。 对于数据生成，我使用以下命令：

from sklearn.datasets import make_moons
X, y = make_moons(n_samples = 100, noise = 0.15)

数据生成后：

X = array([[ 0.83193416,  0.67054039],
       [ 1.4017985 , -0.34708943],
       ...
       [ 1.02640652, -0.58107469],
       [-1.08443914,  0.51960219]])

y = array([0, 1, 1, 1, 0, ... 1, 0])

已生成大小相等的数组，y 是 X 中数据的标签。标记为 0 的数据应显示为圆形，标记为 1 的数据应显示为三角形。 以下是它的外观示例： Example

提前致谢。

Answer 1

使用y 作为选择器来过滤X 的行。绘图的x 和y 向量在X 的*列中，因此我们必须转置数据以使绘图有意义。飞溅星* 用于将列拆分为plot() 的前两个参数。颜色和形状在最终参数 'bo' 和 'g^' 中：

import matplotlib.pyplot as plt

plt.plot(*X[y==0].T, 'bo')  # blue circles (replace o -> s for squares like the picture)
plt.plot(*X[y==1].T, 'g^')  # green triangles
plt.axis('equal')
plt.grid('on')