问题--在matplotlib的散点图中如何给不同种类颜色的点加标签？

---

需求

在进行代码编写的时候，遇到使用散点图的问题，最后呈现出的结果只是不同颜色的点，但是无法一眼看去让人清楚这些点表示的不同含义。因此，通过给散点图中不同颜色的点加标签来表示这些点的含义（我实际的应用是用在机器学习中训练集和测试集的数据中）.

• 原始效果
  

• 最终效果
  

通过该图像中左上角的标签含义可以明白蓝色的散点表示的是训练集的数据，红色的点表示的是测试集的数据。

---

一、scatter图像显示散点含义

• 一元线性回归Python底层实现

• 引入包

import numpy as np
from matplotlib.font_manager import FontProperties
import matplotlib.pyplot as plt
%matplotlib inline

• 构造、计算、测试

# 构造训练的数据集
x_train = [4,8,5,10,12]
y_train = [20,50,30,70,60]

# 画图
# def draw