# 原始流线图
plt.subplot(1, 2, 1)
plt.title("Original Streamlines")
Y, X = np.mgrid[0:h:1, 0:w:1]
plt.streamplot(X, Y, u, -v, density=2, color=magnitude, cmap='viridis') # y方向反转
plt.colorbar(label='Velocity Magnitude')
plt.xlim(0, w)
plt.ylim(0, h)
plt.xlabel('X Position')
plt.ylabel('Y Position')
image.png