这篇文章是python 3.6版本,windows系统下的安装,其他python版本应该也可以,下载对应的包即可。
一定不要直接pip install mayavi,这个玩意儿对vtk的版本有要求。
搞了很久不行,咱也别费那个劲了,直接下whl吧
下载链接: https://www.lfd.uci.edu/~gohlke/pythonlibs/
下载的包:
mayavi-4.7.1+vtk82-cp36-cp36m-win_amd64.whl
VTK-8.2.0-cp36-cp36m-win_amd64.whl
traits-6.1.1-cp36-cp36m-win_amd64.whl
看别的地方说一定要注意安装步骤,读者也可以按下面的安装步骤来
pip install pyqt5 pyqt5-tools
pip install traits-6.1.1-cp36-cp36m-win_amd64.whl
pip install VTK-8.2.0-cp36-cp36m-win_amd64.whl
pip install mayavi-4.7.1+vtk82-cp36-cp36m-win_amd64.whl
代码
import numpy as np
from mayavi import mlab
x, y = np.ogrid[-2:2:20j, -2:2:20j]
z = x * np.exp( - x**2 - y**2)
pl = mlab.surf(x, y, z, warp_scale="auto")
mlab.axes(xlabel='x', ylabel='y', zlabel='z')
mlab.outline(pl)
mlab.show()
这个库是真的不好装,可能读者先看了别人的经验会遇到如下问题:
AttributeError: type object ‘DialogCode’ has no attribute ‘Accepted’
这是因为QT版本的原因(写这个之前看到有人写的安装PyQT4),按这篇文章来安装pyqt5实测不会报这个错。