您可以将箭头自动适合python中的plt文本吗？

如何解决您可以将箭头自动适合python中的plt文本吗？

我有一些带有文本的矩形，这些矩形通过有向边相连，我的代码如下：

import matplotlib.pyplot as plt



plt.close()   

pos_rec={0: (0,0),1:(20,30),2:(50,10)}

text_rec={0:"ShortText",1:"A Bit Longer",2:"A Much Longer Text" }

ax = plt.gca()

edges=((0,1),(1,2),(2,0))


for i in pos_rec:
        plt.text(pos_rec[i][0],pos_rec[i][1],s=text_rec[i],fontsize=15,fontweight="bold",bbox=dict(boxstyle='round',facecolor='red',edgecolor='r',pad=0.7),horizontalalignment='center')


for edge in edges:
    ax.annotate("",xy=pos_rec[edge[0]],xytext=pos_rec[edge[1]],arrowprops=dict(arrowstyle="<|-,head_length=1,head_width=0.5",lw=2,shrinkA=18,shrinkB=20,color='b')
    )
    


 
plt.show()

现在的问题是，边缘被卡在了盒子中，看上去当然并不很漂亮-我可以通过在边缘之后绘制盒子来解决这个问题，因此重叠被盒子隐藏了，但是边缘箭头可能消失;

使用选项rinkaA和rinkbB，我可以在箭头的起点和终点处调整箭头的长度，因此从理论上讲，我可以手动固定箭头的长度，使箭头的确在矩形处结束。问题是这些箭头以不同的深度插在盒子中–而且，根据我选择的比例，这些箭头在这些盒子中的位置可能更深或更浅；

不同比例的代码输出可以在这里看到：

对我来说，很自然的问题是：是否可以使这些边缘直接在这些矩形的边界处终止？

即使独立于缩放比例也可以做到这一点，即无论我放大还是缩小，边缘总是在矩形的边界处结束？

如果这些边缘是弯曲的，如下例所示？

解决方法

您可以使用get_tighbbox()获取边界框，并根据.bounds绘制箭头。建议您阅读matplotlib的transformations tutorial，以获取更多信息。

fig,ax = plt.subplots(1,1,figsize=(7.2,7.2))
ax.set_xlim(0,1)
ax.set_ylim(0,1)

t1 = ax.text(0.5,0.5,'text',bbox=dict(boxstyle='round',facecolor='red',edgecolor='r'))
t2 = ax.text(0.3,0.3,edgecolor='r'))


x1,y1,w1,h1 = t1.get_tightbbox(fig.canvas.get_renderer()).bounds
x2,y2,w2,h2 = t2.get_tightbbox(fig.canvas.get_renderer()).bounds
plt.annotate(' ',(x2+w2,y2+h2),(x1,y1),xycoords='figure pixels',arrowprops=dict(arrowstyle="<|-,head_length=1,head_width=0.5",lw=2,color='b'))