图片集中的传奇故事(python)

Question

我试图在著名的MNIST数据集上可视化不同的流形学习算法.我用文字标记替换了每个数据点，告诉它真正代表的是哪一个数字。但是，我不知道如何添加一个图例，描述属于什么数字的颜色。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import offsetbox
from sklearn import (manifold, datasets, decomposition, ensemble)
import seaborn as sns

digits = datasets.load_digits(n_class=10)
X = digits.data
y = digits.target
n_samples, n_features = X.shape

tsne = manifold.TSNE(n_components=2, random_state=0, perplexity=40)
X_t = tsne.fit_transform(X)

X_t=pd.DataFrame(X_t)
y=pd.DataFrame(y)
XY=pd.concat([X_t,y], axis=1)
XY.columns = ['x', 'y', 'digit']

customPalette=sns.hls_palette(10, l=.7, s=1)


fig, axes = plt.subplots(figsize=(10,10))
axes.set_xlim(XY['x'].min()*0.98,XY['x'].max()*1.02)
axes.set_ylim(XY['y'].min()*0.98,XY['y'].max()*1.02)


for i in range(0,10):
    for l, row in XY.loc[XY['digit']==i,:].iterrows():
        plt.annotate(int(row['digit']), (row['x'], row['y']), 
        horizontalalignment='center', verticalalignment='center', 
        size=10, color=customPalette[i])
        plt.title('t-SNE')

其结果如下：

​

我想在下面的图片中添加一个类似的传说：

图例

Answer 1

像这样的怎么样？

fig, axes = plt.subplots(figsize=(10,10))
axes.set_xlim(XY['x'].min()*0.98,XY['x'].max()*1.02)
axes.set_ylim(XY['y'].min()*0.98,XY['y'].max()*1.02)

for i in range(0,10):
    k=0
    for l, row in XY.loc[XY['digit']==i,:].iterrows():
        if k==0: 
            plt.plot(row['x'], row['y'], '.', color=customPalette[i], label=int(row['digit']))
            plt.legend(numpoints=1, markerscale=3)
            k += 1    
        plt.annotate(int(row['digit']), (row['x'], row['y']), horizontalalignment='center', verticalalignment='center', size=10, color=customPalette[i])
        plt.title('t-SNE')

​

​