Matplotlib绘制指向点箭头

有代码如下:

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()

运行效果:
在这里插入图片描述
  其中,decisionNodeleafNodearrow_args 是定义的三个全局变量,分别表示的是三个基本元素。decisionNodeleafNodecreatePlot() 函数调用 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="<-" 是个反向的箭头

文章内容来源:https://www.jianshu.com/p/438807636568

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值