python编程绘图实例-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()

运行结果：