matplotlib简单和两个头箭头

Question

我想制作一个简单的箭头和一个双头箭头.我使用以下方法制作一个简单的箭头,但我怀疑这是最简单的方法:

import matplotlib.pyplot as plt
arr_width = .009   #  I don't know what unit it is here.
fig = plt.figure()
ax1 = fig.add_subplot(111)
ax1.plot(range(10))
ax1.arrow(1, 1, 0, .5, width = arr_width, head_width = 3 * arr_width, 
    head_length = 9 * arr_width)
plt.show()

我找不到如何用这种方法制作两个头箭.

Answer 1

您可以使用annotate带有空白文本注释的方法创建双头箭头,并将arrowpropsdict 设置为包括arrowstyle='<->'如下所示:

import matplotlib.pyplot as plt

plt.annotate(s='', xy=(1,1), xytext=(0,0), arrowprops=dict(arrowstyle='<->'))

plt.show()

Answer 2

您可以用来matplotlib.patches.FancyArrowPatch绘制双头箭头。该类允许指定arrowstyle：

import matplotlib.patches as patches

p1 = patches.FancyArrowPatch((0, 0), (1, 1), arrowstyle='<->', mutation_scale=20)
p2 = patches.FancyArrowPatch((1, 0), (0, 1), arrowstyle='<|-|>', mutation_scale=20)

这会产生以下箭头：

Answer 3

您可以通过绘制两个plt.arrow重叠的箭头来创建双头箭头。下面的代码有助于做到这一点。

import matplotlib.pyplot as plt

plt.figure(figsize=(12,6))

# red arrow
plt.arrow(0.15, 0.5, 0.75, 0, head_width=0.05, head_length=0.03, linewidth=4, color='r', length_includes_head=True)

# green arrow
plt.arrow(0.85, 0.5, -0.70, 0, head_width=0.05, head_length=0.03, linewidth=4, color='g', length_includes_head=True)

plt.show()

结果是这样的：

您可以看到首先绘制红色箭头，然后绘制绿色箭头。当您提供正确的坐标时，它看起来像一个双头。