在箭头图中添加矩形

Question

我在不同位置(U,V)有一个速度矢量(X,Y)的数据。我知道使用数据绘制箭袋图。

plt.quiver(data[:,0],data[:,1],data[:,3],data[:,4])

在箭袋图中，我需要显示一个矩形区域。

我是python的新手。

Answer 1

matplotlib中的矩形可以使用matplotlib.patches.Rectangle( (x, y), width, height, ...)绘制。

例如，像这样：

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

# generate some data
data = np.random.rand(10,5)
# define a rectangle
# patches.Rectangle( (x, y), width, height, ...)
rectangle = patches.Rectangle( (0.2, 0.3), 0.5, 0.4, alpha=0.5)
ax = plt.gca()
# add the rectangle to the axes
ax.add_patch(rectangle)
# plot the quiver plot
plt.quiver(data[:,0],data[:,1],data[:,2]*5,data[:,3]*5, data[:,4])

plt.show()