有代码如下:
import matplotlib.pyplot as plt
# boxstyle是文本框类型 fc是边框粗细 sawtooth是锯齿形
decisionNode = dict(boxstyle="sawtooth", fc="0.8")
leafNode = dict(boxstyle="round4", fc="0.8")
arrow_args = dict(arrowstyle="<-")
# annotate 注释的意思
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('a decision node', (0.5, 0.1), (0.1, 0.5), decisionNode)
plotNode('a leaf node', (0.8, 0.1), (0.3, 0.8), leafNode)
plt.show()
createPlot()
运行效果:
其中,decisionNode 、leafNode 和 arrow_args 是定义的三个全局变量,分别表示的是三个基本元素。decisionNode 和 leafNode 在 createPlot() 函数调用 plotNode() 的时候作为参数。arrow_args 是用来画箭头的
- plotNode():plotNode函数有nodeTxt,centerPt, parentPt, nodeType这四个参数。nodeTxt用于记录nodeTxt,即节点的文本信息。centerPt表示那个节点框的位置。 parentPt表示那个箭头的起始位置。nodeType表示的是节点的类型,也就会用我们之前定义的全局变量。
- createPlot.ax1:在createPlot中第三行createPlot.ax1 = plt.subplot(111, frameon=False) 进行了创建。意思是这个只是一个新框。annotate是注释的意思,也就是作为原来那个框的注释,也是添加一些新的东西。
对于 annotate 的用法,可参考以下博客:
https://blog.csdn.net/wizardforcel/article/details/54782628
https://blog.csdn.net/leaf_zizi/article/details/82886755
参数意义:
boxstyle是文本框类型
- sawtooth是锯齿图文框
- round4是圆一点的四边形(见图)
arrowstyle是设置箭头类型
- 文档显示 Axes.annotate(*args, **kwargs)
- xy是终点坐标
- xytext是起点坐标
注意:arrowstyle="<-" 是个反向的箭头