python mayavi_python mayavi三维绘图

网上下载mayavi的官方帮助文档，里面有很多例子，下面的记录都是查看手册后得到的。

Python的mayavi.mlab库中的绘图函数有很多候选参数，但下文记录并没有过多讨论，本人也是需要用到才查看手册的。

安装好mayavi2的绘图环境后，可以结合numpy进行科学绘图，在代码中事先加入如下代码：

import mayavi.mlab as mlab

from numpy import exp,sin,cos,tan,random,mgrid,ogrid,linspace,sqrt,pi

import numpy as np

import matplotlib.pyplot as plt

mlab.figure(fgcolor=(0, 0, 0), bgcolor=(1, 1, 1)) #更改背景色

#添加matlab的peaks函数

def peaks(x,y):

return 3.0*(1.0-x)**2*exp(-(x**2) - (y+1.0)**2) - 10*(x/5.0 - x**3 - y**5) * exp(-x**2-y**2) - 1.0/3.0*exp(-(x+1.0)**2 - y**2)1

2

3

4

5

6

7

8

91

2

3

4

5

6

7

8

9

首先从帮助手册上了解下mayavi的colormap，如下图：

下面列举常用的三维绘图函数和简单例子。

一、barchart

* barchart(s, ...)

* barchart(x, y, s, ...)

* barchart(x, y, f, ...)

* barchart(x, y, z, s, ...)

* barchart(x, y, z, f, ...)

如果只传递一个参数，可以是一维(1-D)，二维(2-D)或3维(3-D)的给定向量长度的数组；

如果传递三个参数(x,y,s)或(x,y,f),x,y是对应于数组s的二维(2-D)坐标，也可以是可调用的函数f，该函数返回数组；

四个参数的时候(x,y,z)表示三维坐标

s = np.random.rand(3,3)

mlab.barchart(s)

mlab.vectorbar()

mlab.show()1

2

3

41

2

3

4

x,y = np.mgrid[-5:5:20j,-5:5:20j]

s = peaks(x,y) #peaks函数前面已经定义

mlab.barchart(x,y,s)

mlab.vectorbar()

mlab.show()1

2

3

4

51

2

3

4

5