用matplotlib绘制三维概率分布

Question

我有一个1024x1024x1024的矩阵，这个矩阵就像一个三维空间，每个单元格对应空间中的一个点。这个矩阵里的每个元素都有一个值，表示在那个坐标点上粒子出现的概率。

我想用matplotlib来绘制一个三维的概率分布。我可以用imshow在二维中做到这一点，但我不知道怎么在三维中实现。

谢谢

Answer 1

可能你想要的是MayaVi的三维点数据：

#!/usr/bin/env python2

import numpy as np
from mayavi import mlab

x, y, z, value = np.random.random((4, 40))
mlab.points3d(x, y, z, value)

mlab.show() # or
# mlab.savefig("1.png", size=(1000,800))

结果：