使用来自文件

Question

我做了很多事情来绘制流线型，并做到了这一点。我asked关于矢量场并找到了答案。现在我想用matplotlib绘制流线。现在，我已经有了用于在gnuplot中绘制矢量场的输出数据。我已将数据输出为X Y U V，并希望将其用于简化流程。

而且我不知道如何为图像设置输出首选项。我有此示例代码。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

Y, X = np.mgrid[0:1:31j, 0:1:31j]
U = -1 - X**2 + Y
V = 1 + X - Y**2
speed = np.sqrt(U**2 + V**2)

fig = plt.figure(figsize=(7, 9))
gs = gridspec.GridSpec(nrows=3, ncols=2, height_ratios=[1, 1, 2])

# Varying color along a streamline
ax1 = fig.add_subplot(gs[0, 1])
strm = ax1.streamplot(X, Y, U, V, color=U, linewidth=1, cmap='gnuplot')
fig.colorbar(strm.lines)
ax1.set_title('Navier-Stokes Equation')


plt.tight_layout()
fig.savefig('streamlines.png')
plt.show()

要进行哪些编辑以从文件输入U和V并将其输出为格式化图像？