1. 导入需要使用的库:numpy、matplotlib.pyplot、LinearSegmentedColormap。
  2. 设置中文字体。
  3. 创建一个画布和一个子图,并设置子图的背景颜色为浅灰色。
  4. 设置x和y的最小值和最大值。
  5. 使用meshgrid函数生成一个网格,xx和yy分别表示每个点的横坐标和纵坐标。
  6. 使用随机森林模型对网格数据进行预测,并将预测结果Z重塑为与xx和yy相同的形状。
  7. 创建一个颜色映射,将分类为-1的点的颜色设置为红色,将分类为1的点的颜色设置为蓝色。
  8. 使用pcolormesh函数将预测结果可视化,将x轴和y轴的范围设置为xx和yy的最小值和最大值。
  9. 获取标签为-1和1的数据点的横坐标和纵坐标,并将其分别存储在x1、y1和x2、y2中。
  10. 使用scatter函数将标签为-1和1的数据点可视化,将标签为-1的点的颜色设置为红色圆圈,将标签为1的点的颜色设置为蓝色叉号。
  11. 设置子图的标题、x轴标签和y轴标签,并将它们的颜色设置为深蓝色。
  12. 设置刻度的标签颜色为深蓝色。
  13. 使用legend函数添加图例,并将红色圆圈标记为“-1”,将蓝色叉号标记为“1”。
  14. 显示图形
import numpy as npimport matplotlibpyplot as pltfrom matplotlibcolors import LinearSegmentedColormapmatplotlib notebookpltrcParamsfontsans-serif = PingFang HK # 选择一个本地的支持中文的字体fig ax = pltsubplotsaxse

原文地址: https://www.cveoy.top/t/topic/hng0 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录