Python绘图模板：流向图源代码精简版.pdfVIP

流向图

importnumpyasnpimport

matplotlib.pyplotaspltimport

matplotlib.gridspecasgridspec

w=3Y,X=np.mgrid[‑w:w:100j,‑

w:w:100j]U=‑1‑X**2+YV=1+

X‑Y**2速度=np.sqrt(U*U+

V*V)

fig=plt.figure(figsize=(7,9))gs=gridspec.GridSpec(nro

=3_,1,2])

ws,ncols=2,heightratios=[1

#流线密度变化ax0=fig.add_

subplot(gs[0,0])ax0.streamplot(X,

Y,U,V,density=[0.5,1])ax0.set_

title(流线密度变化)

_

#流线颜色变化ax1=fig.addsubplot(gs[0,1])strm=

ax1.streamplot(X,Y,U,V,color=U,linewidth=2,cmap