Plot 使用UMAP进行聚类和绘图

我正在尝试使用UMAP研究一些数据集的形状。我所有的数据都在三个矩阵A_1，A_2，A_3中，它们包含相同维度的向量

我想使用UMAP将它们转换在一起，并以一种打印颜色随所属矩阵而变化的方式打印它们。换句话说，我希望能够区分这三个类

    fit = umap.UMAP()
    %time u = fit.fit_transform(A)

    plt.scatter(u[:,0], u[:,1])

---

给了我关于A的表示，我试过把向量连接到一个唯一的矩阵中，并且成功了。唯一的问题是，到了那个时候，我不知道如何区分情节中的点。如何解决这个问题？

通过索引来区分它们

即，通过索引查找它们、项目、un concat

就像处理标签列一样