《机器学习实战》学习笔记四：使用Matplotlib注解绘制树形图

最新推荐文章于 2024-07-26 11:40:56 发布

愛在梦醒时分

最新推荐文章于 2024-07-26 11:40:56 发布

阅读量925

点赞数 1

分类专栏： python 机器学习《机器学习实战》学记笔记

本文链接：https://blog.csdn.net/h843653844/article/details/79155625

版权

本文是《机器学习实战》的学习笔记，主要介绍了如何利用Matplotlib的注解功能绘制树形图。首先，讲解了如何用文本注解绘制树节点，然后通过两个辅助函数确定树的结构。接着，代码示例展示了如何完善树的绘制，包括设置尺寸、位置、x轴和y轴标签，最终得到清晰的树形图。

摘要由CSDN通过智能技术生成

Matplotlib是python的一个绘图库，可以使用其注解功能来绘制图形。
1.使用文本注解绘制树节点

import matplotlib.pyplot as plt
decisionNode = dict(boxstyle="sawtooth",fc="0.8")#定义文本框和箭头格式
leafNode = dict(boxstyle="round4",fc="0.8")
arrow_args = dict(arrowstyle="<-")
def plotNode(nodeTxt,centerPt,parentPt,nodeType):#绘制带箭头的注解
    createPlot.axl.annotate(nodeTxt,xy=parentPt,xycoords='axes fraction',xytext= centerPt,\
    textcoords='axes fraction',va="center",ha="center",bbox=nodeType,arrowprops=arrow_args)
def createPlot():
    fig = plt.figure(1,facecolor='white')