流线图与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()