Python 用颜色渐变分离类绘制图形

我的数据包含3列或特征，即X、Y、class，如下所示

  X     Y   Colour
  0     0   Blue
  0     0.5 Blue
  0     1   Blue
  0.2   1   Blue
  0.4   2   Red
  0.4   2.5 Red
  .     
  .
  .

我绘制了一个散点图，其中X和Y参数以及第三个参数作为类，如下所示

但我想通过decisionboundary来分离类，并希望类似这样的东西，我使用paint手动创建的。

---

请帮助我尝试使用python matplotlib和逻辑回归来创建分离边界。

一个选项是使颜色列为数字，并调用

plt.tourf（）

或

plt.tricontourf（）

。看见e、 g.一个选项是使颜色列为数字，并调用

plt.contourf（）

或

plt.tricontourf（）

。看见例如