Python matplotlib绘图示例 - 绘制三维图形

Python matplotlib模块是扩展的MATLAB的一个绘图工具库。它可以绘制各种图形，下面就学习了下Python中的matplotlib模块，如何绘制三维图形。

示例代码一：

# coding=utf-8
import numpy as np
import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d
a,b = np.mgrid[-2:2:20j,-2:2:20j]
#测试数据
c=a*np.exp(-a**2-b**2)
#三维图形
ax = plt.subplot(111, projection='3d')
ax.set_title('www.linuxidc.com - matplotlib Code Demo');
ax.plot_surface(a,b,c,rstride=2, cstride=1, cmap=plt.cm.Spectral)
#设置坐标轴标签
ax.set_xlabel('A')
ax.set_ylabel('B')
ax.set_zlabel('C')
plt.show()

运行结果：

示例代码二：
# coding=utf-8
import pylab as pl
import numpy as np
import mpl_toolkits.mplot3d
rho, theta = np.mgrid[0:1:40j,0:2*np.pi:40j]
c = rho**2
a = rho*np.cos(theta)
b = rho*np.sin(theta)
ax = pl.subplot(111, projection='3d')
ax.set_title('www.linuxidc.com - matplotlib Code Demo');
#ax.plot_surface(a,b,c)
ax.plot_surface(a,b,c,rstride=2, cstride=1)
#设置坐标轴标签
ax.set_xlabel('A')
ax.set_ylabel('B')
ax.set_zlabel('C')
pl.show()

运行结果：