python - 使用 matplotlib 绘制 3D 概率分布

Question

我有一个 1024x1024x1024 矩阵，它类似于 3D 空间，每个单元格对应于空间中的一个点。该矩阵的每个元素都有一个值，该值决定了该坐标中粒子存在的概率。

我想用 matplotlib 绘制一个三维概率分布。我在 2D 中使用 imshow 执行此操作，但我不知道如何在 3D 中制作它。

谢谢

Answer 1

可能你想要 MayaVi 的 points3d：

#!/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))

结果：