matplotlib函数库使用Axes3D绘制3D图形

来源:互联网 发布:c语言源码网站 编辑:程序博客网 时间:2024/06/03 21:54
#coding=utf-8import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3Dfig=plt.figure()ax=Axes3D(fig)x=np.arange(-5,5,0.25)y=np.arange(-5,5,0.25)x,y=np.meshgrid(x,y)r=np.sqrt(x**2+y**2)z=np.sin(r)#高度ax.plot_surface(x,y,z,rstride=1,cstride=1,cmap=plt.get_cmap('rainbow'))#填充rainbow颜色ax.contourf(x,y,z,zdir='z',offset=-2,cmap='rainbow')#绘制3D图形,zdir表示从哪个坐标轴上压下去plt.show()#显示图片

原创粉丝点击