# 原始流線圖
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方向反轉(zhuǎn)
plt.colorbar(label='Velocity Magnitude')
plt.xlim(0, w)
plt.ylim(0, h)
plt.xlabel('X Position')
plt.ylabel('Y Position')

image.png