流水图不匹配mgrid

Question

我正在使用Python代码绘制2D ODE系统的流图：

import numpy as np
import matplotlib.pyplot as plt

Y, X = np.mgrid[-4:4:100j, -4:4:100j]

U =  -0.5*X - Y
V =  X - Y
plt.streamplot(X, Y, U, V, density=[1, 1])
plt.show()

然而，我得到了下面的图，它将x和y轴扩展为一个额外的单元，5而不是4，即使mgrid在两个方向上都是-4到4。我试过不同的范围，但没有成功。任何关于如何将情节限制在-4:-4和-4:4的想法

​

​

Answer 1

你可以用

ax.set(xlim=(-4,4), ylim=(-4,4))

若要手动设置限制，请执行以下操作：

import numpy as np
import matplotlib.pyplot as plt

fig, ax = plt.subplots()
Y, X = np.mgrid[-4:4:100j, -4:4:100j]
U =  -0.5*X - Y
V =  X - Y
ax.streamplot(X, Y, U, V, density=[1, 1])
ax.set(xlim=(-4,4), ylim=(-4,4))
plt.show()

​

​