FancyArrowPatch 不绘制标签

Question

我无法让 FancyArrowPatch 显示传递给 label kwarg 的文本。我看不出我做错了什么。有什么想法吗？

设置：

import matplotlib.pyplot as plt
from matplotlib import patches
import numpy as np

fig = plt.figure()

X, Y = np.mgrid[-1:1:.1, -1:1:.1]
Z = X+Y

ax1 = fig.add_subplot(311)
ax2 = fig.add_subplot(313)
ax1.contourf(X, Y, Z)
ax2.contourf(X, Y, -Z)

箭头代码：

ax1tr = ax1.transData # Axis 0 -> Display
ax2tr = ax2.transData # Axis 1 -> Display
figtr = fig.transFigure.inverted() # Display -> Figure
ptB = figtr.transform(ax1tr.transform((0., 0.)))
ptE = figtr.transform(ax2tr.transform((0., 0.)))
arrow = patches.FancyArrowPatch(
    ptB, ptE, transform=fig.transFigure,  # Place arrow in figure coord system
    connectionstyle="arc3,rad=0", arrowstyle='simple',
    mutation_scale = 15., label='some arrow'
)
fig.patches.append(arrow)

fig.show()

我希望 some arrow 被绘制为箭头的标签。但事实并非如此。

Answer 1

原则上它应该在您的代码中添加plt.legend() 行后工作。但是，有一个解决方法如下，您可以显式传递arrow 补丁和标签文本。

arrow = patches.FancyArrowPatch(
    ptB, ptE, transform=fig.transFigure,  # Place arrow in figure coord system
    connectionstyle="arc3,rad=0", arrowstyle='simple',
    mutation_scale = 15., label='some arrow'
)
fig.patches.append(arrow)

plt.legend([arrow], fontsize=16)

另一种选择是

plt.legend(handles=[arrow], fontsize=16)

基于 cmets，正如我所说，label 的目的不是将文本放在箭头补丁旁边。使用plt.text 可以轻松实现您想要的。但是，如果您不想使用它，可以使用以下解决方法使当前代码按需要工作

leg = plt.legend(handles = [arrow], fontsize=16, frameon=False, loc=(0.45, 0.5))

for item in leg.legendHandles:
    item.set_visible(False)