如果要使用颜色图,则可以为
colors列表中的每个唯一条目创建图例条目,如下所示。这种方法适用于任何数量的值。图例手柄是的标记
plot,因此它们与分散点匹配。
import matplotlib.pyplot as pltx_values = [6.2, 3.6, 7.3, 3.2, 2.7]y_values = [1.5, 3.2, 5.4, 3.1, 2.8]colors = [1, 1, 0, 1, -1]labels = ["a", "a", "b", "a", "c"]clset = set(zip(colors, labels))ax = plt.gca()sc = ax.scatter(x_values, y_values, c=colors, cmap="brg")handles = [plt.plot([],color=sc.get_cmap()(sc.norm(c)),ls="", marker="o")[0] for c,l in clset ]labels = [l for c,l in clset]ax.legend(handles, labels)plt.show()



