Построение 3D-точек с помощью Python Matploltlib

Question

Я хочу создать небольшую симуляцию, и я думаю, что знаю как, но для того, чтобы действительно увидеть, что происходит, мне нужно визуализировать ее.

Я начал с массива 5x5x5, который я хочу заполнить значениями.

data = numpy.zeros(shape=(5,5,5)) 
data[:,:,0]=4 
data[:,:,1]=3 
data[:,:,2]=2 
data[:,:,3]=1 
data[:,:,4]=0 

Это должно создать куб с градиентом в направлении вверх (если третья ось равна z).

Теперь, как я могу построить это? Мне не нужен поверхностный сюжет или каркас. Просто точки на каждой координате и, возможно, цветные или трансперенсивные по стоимости.

В качестве теста я пытался черчения все координаты с помощью

ax.scatter(numpy.arange(5),numpy.arange(5),numpy.arange(5)) 

, но это будет только построить линию, состоящую из 5 точек.

Итак ... как я могу получить 125 точек, которые я хочу создать?

Thx.

Answer 1

Вы можете кодировать значение в цвете, как это:

x = np.arange(5) 
X, Y, Z = np.meshgrid(x,x,x) 

v = np.arange(125) 
ax.scatter(X,Y,Z, c=v) 

here См для справки о.