用plt.scatter()画带label的散点图（无需循环，直接根据标签生成）

前言

网上找了很多例子都没找到我想要的，最后翻官方文档找到了方法

代码

官方文档
拿4个点举个例子：

fig, ax = plt.subplots()
scatter = ax.scatter(x,y, marker='.', c=[1,2,3,4])
legend1 = ax.legend(*scatter.legend_elements(),loc="lower left", title="Classes")
ax.add_artist(legend1)

通过观察scatter.legend_elements()观察a和b

a,b=scatter.legend_elements()
print(b)

把b修改为

b=['$\\mathdefault {type A}$',
 '$\\mathdefault{type B}$',
 '$\\mathdefault{type C}$',
 '$\\mathdefault{type D}$']
legend1 = ax.legend(a,b, title="Classes")

就可以得到