python - 如何在matplotlib中使用 `quiver()`获取单位向量？

Question

我正在尝试用quiver函数来绘制向量场。这是一个测试用例:

import numpy as np
import matplotlib.pyplot as plt
X, Y = np.mgrid[1:1.5:0.5, 1:1.5:0.5]
print(X)
print(Y)
u = np.ones_like(X)
v = np.zeros_like(Y)
plt.quiver(X,Y, u, v)
plt.axis([0, 3, 0, 3], units='xy', scale=1.)
plt.show()

我试图获取长度为 1 的向量，从 (1,0) 指向 (2,0)，但这是我得到的:

我尝试添加 scale='xy' 选项，但行为没有改变。那么这是如何工作的呢？

Answer 1

第一个有趣的错误是您将 quiver 参数放入 axis 调用中。 ;-)

接下来，查看the documentation ，它说

If scale_units is ‘x’ then the vector will be 0.5 x-axis units. To plot vectors in the x-y plane, with u and v having the same units as x and y, use angles='xy', scale_units='xy', scale=1.

所以让我们按照文档告诉我们的去做，

import numpy as np
import matplotlib.pyplot as plt
X, Y = np.mgrid[1:1.5:0.5, 1:1.5:0.5]

u = np.ones_like(X)
v = np.zeros_like(Y)
plt.quiver(X,Y, u, v, units='xy', angles='xy', scale_units='xy', scale=1.)
plt.axis([0, 3, 0, 3])
plt.show()

我们确实得到了一个单位长箭头: