在散点之间绘制多个箭头

Question

我正在尝试在两组散点之间绘制多个箭头。使用 ax.plot 绘制一条线很容易。但我正在尝试实现一个箭头而不是一条线。箭头似乎没有在点之间对齐。

因此，如果在下面初始化线图，它就可以正常工作。但是箭袋图并不单独绘制相同的线。

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

df = pd.DataFrame(np.random.randint(-50,50,size=(100, 4)), columns=list('ABCD'))

fig, ax = plt.subplots()

x1 = df['A']
y1 = df['B']

x2 = df['C']
y2 = df['D']

AB = plt.scatter(x1, y1, c = 'blue', marker = 'o', s = 10, zorder = 3)
CD = plt.scatter(x2, y2, c = 'red', marker = 'o', s = 10, zorder = 2)

# plot line between points
#ax.plot([x1,x2],[y1,y2], color = 'black', linestyle = '--', linewidth = 0.5)

ax.quiver([x1, x2], [y1, y2])

Answer 1

根据documentation，见scale_units选项，你需要：angles='xy', scale_units='xy', scale=1 in quiver:

AB = ax.scatter(x1, y1, c = 'blue', marker = 'o', s = 10, zorder = 3)
CD = ax.scatter(x2, y2, c = 'red', marker = 'o', s = 10, zorder = 2)

ax.quiver(x1, y1, (x2-x1), (y2-y1), angles='xy', scale_units='xy', scale=1)
plt.show()

输出：

在散点之间绘制多个箭头

plot multiple arrows between scatter points

python

matplotlib

arrow-functions