python - 是否可以使用列标签在 T-SNE 中定义点？

Question

我有 67382 行数据的 pandas DataFrame。其中目标 - 时间戳中的日期时间。

我创建了用于绘图、计算 T-sne 的代码，并在下面的代码中显示了 T-sne 点：

from sklearn.manifold import TSNE
features = ['TDS7,C', 'TDS8,C', 'TDS9,C', 'target']
tsne = TSNE(n_components=2, n_iter=1000, random_state=RANDOM_STATE)
points = tsne.fit_transform(df[features])
plt.scatter(points[:0], points[:1], c=df['target'], cmap=sns.color_palette("crest", as_cmap=True))
plt.title(title, fontsize=20, y=1.03)
plt.xlabel(xlabel, fontsize=16)
plt.ylabel(ylabel, fontsize=16)

但是，可以将列的标签映射到点（确定哪个点到哪个列）和图例，如下图所示？例如：

其中，人口 A - TDS7，人口 B - TDS8，人口 C - TDS9。