Python 如何将Lenged放入matplotlib，将数据放在同一个numpy数组中

我正在阅读《机器学习在行动》第二章，我不知道如何标记分为三组的点

数据有四列，前三列中的两列用作x和y，最后一列是标签（1、2和3）

数据集位于同一numpy数组中，并使用.scatter（X，Y，15*标签值，15*标签值）以不同的颜色和大小打印

以下是无图例打印的代码

fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(datingDataMat[:, 0], datingDataMat[:, 1], 15 * np.array(datingLabels), 15 * np.array(datingLabels))
fig.suptitle('Figure 2 Using X = Colume 0 and Y = Colume 1')
plt.xlabel('Frequent Flyier Miles Earned Per Year')
plt.ylabel('Percentage of Time Spent Playing Video Games')

plt.show()

---

不一样。数据来自同一阵列，并在同一散点中以不同大小绘制。我想问一下如何实现这一不同的标签。