ML基础-决策树-4-构造注解树

绘制树节点

import matplotlib.pyplot as plt

# ❶ （以下三行）定义文本框和箭头格式
decisionNode = dict(boxstyle="sawtooth", fc="0.8")
leafNode = dict(boxstyle="round4", fc="0.8")
arrow_args = dict(arrowstyle="<-")


# ❷ （以下两行）绘制带箭头的注解
# xycoords为目标点的坐标系统,axes fraction指按照坐标轴相对位置
# xytext在坐标为xytext处添加坐标为x, y的点的注释
# textcoords为目标点的坐标系统, axes fraction指按照坐标轴相对位置
# bbox应该是（矩形的）文本框属性，alpha是相对不透明度，facecolor应该是背景颜色
# arrowprops的属性：shrink, facecolor, width（箭头宽度），headwidth（箭头宽度)
def plotNode(nodeTxt, centerPt, parentPt, nodeType):
    createPlot.ax1.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')
    fig.clf()
    createPlot.ax1 = plt.subplot(111, frameon=False)
    plotNode('决策节点', (0.5, 0.1), (0.1</