14. Mayavi可视化实例

Dragon绘制

import shutil
import tarfile
from mayavi import mlab
from os.path import join

# 读取tar压缩文件
dragon_tar_file = tarfile.open('Data/dragon.tar.gz')
dragon_tar_file.extractall('dragon_data')
dragon_tar_file.close()
dragon_ply_file = join('dragon_data', 'dragon_recon', 'dragon_vrip.ply')

# 渲染dragon ply文件
mlab.pipeline.surface(mlab.pipeline.open(dragon_ply_file))
mlab.show()

# 删除解压的文件夹
shutil.rmtree('Data/dragon_data')

Canyon地形可视化

import zipfile
import numpy as np
from mayavi import mlab

# 读取压缩文件
# N36W113.hgt：北纬36-37度，西经113-114度 地形高程数据。
hgt = zipfile.ZipFile('Data/N36W113.hgt.zip').read('N36W113.hgt')
data = np.fromstring(hgt, '>i2')
# 转换数据类型
data.shape = (3601, 3601)
data = data.astype(np.float32)
# 选取一部分数据
data = data[:1000, 900:1900]
# data == -32768代表空缺值
# 令空缺值等于最小值
data[data == -32768] = data[data