3D数据

 1 import matplotlib.pyplot as plt
 2 import numpy as np
 3 from mpl_toolkits.mplot3d import Axes3D
 4 
 5 fig = plt.figure()
 6 ax = Axes3D(fig)
 7 
 8 #X,Y value
 9 X = np.arange(-4,4,0.25)
10 Y = np.arange(-4,4,0.25)
11 X,Y = np.meshgrid(X,Y)
12 R = np.sqrt(X**2+Y**2)
13 #height value
14 Z = np.sin(R)
15 
16 ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap=plt.get_cmap('rainbow'))
17 
18 plt.show()

运行结果如下：

 

 1 import matplotlib.pyplot as plt
 2 import numpy as np
 3 from mpl_toolkits.mplot3d import Axes3D
 4 
 5 fig = plt.figure()
 6 ax = Axes3D(fig)
 7 
 8 #X,Y value
 9 X = np.arange(-4,4,0.25)
10 Y = np.arange(-4,4,0.25)
11 X,Y = np.meshgrid(X,Y)
12 R = np.sqrt(X**2+Y**2)
13 #height value
14 Z = np.sin(R)
15 
16 ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap=plt.get_cmap('rainbow'))
17 
18 ax.contourf(X,Y,Z,zdir='z',offset=-2,cmap='rainbow') #从Z轴这个方向进行映射到x0y平面
19 ax.set_zlim3d(-2,2)
20 
21 plt.show()

运行结果如下：