python中图例legend标签内容_在python中采用plt.text形式描点绘图时，怎么通过legend添加图例？...

调用的代码块如下：#coding='utf-8'"""t-SNE对手写数字进行可视化"""fromtimeimporttimeimportnumpyasnpimportmatplotlib.pyplotaspltfromsklearnimportdatasetsfromsklearn.manifold...

调用的代码块如下：# coding='utf-8'"""t-SNE对手写数字进行可视化"""from time import timeimport numpy as npimport matplotlib.pyplot as pltfrom sklearn import datasetsfrom sklearn.manifold import TSNEdef get_data(): digits = datasets.load_digits(n_class=6) data = digits.data label = digits.target n_samples, n_features = data.shape return data, label, n_samples, n_featuresdef plot_embedding(data, label, title): x_min, x_max = np.min(data, 0), np.max(data, 0) data = (data - x_min) / (x_max - x_min) fig = plt.figure() ax = plt.subplot(111) for i in range(data.shape[0]): plt.text(data[i, 0], data[i, 1], str(label[i]), color=plt.cm.Set1(label[i] / 10.), fontdict={'weight': 'bold', 'size': 9}) plt.xticks([]) plt.yticks([]) plt.title(title) return figdef main(): data, label, n_samples, n_features = get_data() print('Computing t-SNE embedding') tsne = TSNE(n_components=2, init='pca', random_state=0) t0 = time() result = tsne.fit_transform(data) fig = plot_embedding(result, label, 't-SNE embedding of the digits (time %.2fs)' % (time() - t0)) plt.show(fig)if __name__ == '__main__': main()这是一个通过t-SNE降维可视化的代码块，结果如图1所示。我想在此图右上角添加图例，但使用ax.legend()代码添加图例失败，请问针对此例正确添加图例的代码是什么？（理想图例如图2所示）

代码上传之后换行符被自动去掉了很乱，这是该代码块的网页链接：网页链接

展开