python - 流图与 mggrid 不匹配

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，即使 mggrid 在两个方向上都是 -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()