python - 如何使用 matplotlib 将 3d 矩阵映射到 3d 散点图中的颜色值？

Question

我有一个 10x10x10 numpy 矩阵，我试图在 3d 中进行可视化:

from mpl_toolkits.mplot3d import Axes3D
M = np.random.rand(10, 10, 10)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
counter = range(10)
ax.scatter(counter, counter, counter, c=??)

我想要一个 3d 图，其中位置 i,j,k 的黑暗度由 M[i,j,k] 给出。我应该如何将 M 传递给 scatter() 以便它正确执行此操作？它似乎需要一个二维数组，但我不明白在这种情况下它是如何工作的。

Answer 1

散点图需要与颜色数组 c 相同数量的点。因此，对于 1000 种颜色，您需要 1000 点。

import matplotlib.pyplot as plt
import numpy as np

from mpl_toolkits.mplot3d import Axes3D
M = np.random.rand(10, 10, 10)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
counter = range(10)
x,y,z = np.meshgrid(counter, counter, counter)
ax.scatter(x,y,z, c=M.flat)


plt.show()