如何使用matplotlib将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()

Answer 2

尝试：

scatter3D(xs, ys, zs=0, zdir='z', s=20, c=None, depthshade=True, *args, **kwargs)

但是c！=没有

https://matplotlib.org/mpl_toolkits/mplot3d/api.html