6.2 数组和区域绘图

6.2.1 imshow(Z)

import matplotlib.pyplot as plt
import numpy as np

#make data
X,Y=np.meshgrid(np.linspace(-3,3,16),np.linspace(-3,3,16))
Z=(1-X/2+X**5+Y**3)*np.exp(-X**2-Y**2)

#plot
fig=plt.figure()
ax=plt.subplot()

ax.imshow(Z)
plt.show()

6.2.2 pcolormesh(X,Y,U,V)

import matplotlib.pyplot as plt
import numpy as np

#make data with uneven sampling in x
x=[-3,-2,-1.6,-1.2,-.8,-.5,-.2,.1,.3,.5,.8]
X,Y=np.meshgrid(x,np.linspace(-3,3,128))
Z=(1-X/2+X**5+Y**3)*np.exp(-X**2-Y**2)

#plot
fig=plt.figure()
ax=plt.subplot()

ax.pcolormesh(X,Y,Z,vmin=-0.5,vmax=1.0)
plt.show()

6.2.3 contour(X,Y,Z)

import matplotlib.pyplot as plt
import numpy as np

#make data
X,Y=np.meshgrid(np.linspace(-3,3,256),np.linspace(-3,3,256))
Z=(1-X/2+X**5+Y**3)*np.exp(-X**2-Y**2)
levels=np.linspace(np.min(Z),np.max(Z),7)

#plot
fig=plt.figure()
ax=plt.subplot()

ax.contour(X,Y,Z,levels=levels)
plt.show()

6.2.4 contourf(X,Y,Z)

import matplotlib.pyplot as plt
import numpy as np

#make data
X,Y=np.meshgrid(np.linspace(-3,3,256),np.linspace(-3,3,256))
Z=(1-X/2+X**5+Y**3)*np.exp(-X**2-Y**2)
levels=np.linspace(Z.min(),Z.max(),7)

#plot
fig=plt.figure()
ax=plt.subplot()

ax.contourf(X,Y,Z,levels=levels)

plt.show()

6.2.5 barbs(X,Y,U,V)

import matplotlib.pyplot as plt
import numpy as np

#make data
X,Y=np.meshgrid(np.linspace(-3,3,256),np.linspace(-3,3,256))
Z=(1-X/2+X**5+Y**3)*np.exp(-X**2-Y**2)
levels=np.linspace(Z.min(),Z.max(),7)

#plot
fig=plt.figure()
ax=plt.subplot()

ax.contourf(X,Y,Z,levels=levels)

plt.show()

6.2.6 quiver(X,Y,U,V)

import matplotlib.pyplot as plt
import numpy as np

#make data
X,Y=np.meshgrid([1,2,3],[1,2,3,4])
angle=np.pi/180*np.array([[10,20,30,60],
[25,40,55,60],
[35,50,65,75]])
amplitude=np.array([[5,10,25,50],
[10,15,30,60],
[15,26,50,70]])

U=amplitude*np.sin(angle)
V=amplitude*np.cos(angle)

#plot
fig=plt.figure()
ax=plt.subplot()

ax.barbs(X,Y,U,V,barbcolor="C0",flagcolor="C0",length=7,linewidth=1.5)

plt.show()

6.2.7 streamplot(X,Y,U,V)

import matplotlib.pyplot as plt
import numpy as np

#make data
x=np.linspace(-4,4,6)
y=np.linspace(-4,4,6)
X,Y=np.meshgrid(x,y)
U=X+Y
V=X-Y

#plot
fig=plt.figure()
ax=plt.subplot()

ax.quiver(X,Y,U,V,color="C0",angles="xy",
scale_units="xy",scale=5,width=0.015)

plt.show()