Определение цветов графика штриховки Matplotlib 3D

Question

Я не могу найти правильный способ установить cmap (или цвета) для 3D-графика в matplotlib в моем ноутбуке iPython. Я могу правильно настроить диаграмму (28 x 7 меток) в плоскости X и Y с некоторыми случайными значениями Z. График трудно интерпретировать, и одна из причин заключается в том, что цвета по умолчанию для ярлыков x_data [1,2,3,4,5] одинаковы.

Вот код:

%matplotlib inline 
from mpl_toolkits.mplot3d import Axes3D 
import matplotlib.pyplot as plt 
import numpy as npfig = plt.figure(figsize=(18,12)) 

ax = fig.add_subplot(111, projection='3d') 

x_data, y_data = np.meshgrid(np.arange(5),np.arange(3)) 
z_data = np.random.rand(3,5).flatten() 

ax.bar3d(x_data.flatten(), 
y_data.flatten(),np.zeros(len(z_data)),1,1,z_data,alpha=0.10) 

Который производит следующую таблицу:

Я не хочу, чтобы определить цвета вручную для x_data этикеток. Как я могу настроить различные «случайные» CMAP цвета для каждой из меток в x_data, все еще держа

ax.bar3d

параметр? Ниже приведен вариант с использованием

ax.bar

и различные цвета, но то, что мне нужно, это ax.bar3d.

Answer 1

from mpl_toolkits.mplot3d import Axes3D 
import matplotlib.pyplot as plt 
import numpy as np 

fig = plt.figure(figsize=(18,12)) 
ax = fig.add_subplot(111, projection='3d') 

x_data, y_data = np.meshgrid(np.arange(5),np.arange(3)) 
z_data = np.random.rand(3,5) 
colors = ['r','g','b'] # colors for every line of y 

# plot colored 3d bars 
for i in xrange(3): # cycle though y 
    # I multiply one color by len of x (it is 5) to set one color for y line 
    ax.bar3d(x_data[i], y_data[i], z_data[i], 1, 1, z_data[i], alpha=0.1, color=colors[i]*5) 
    # or use random colors 
    # ax.bar3d(x_data[i], y_data[i], z_data[i], 1, 1, z_data[i], alpha=0.1, color=[np.random.rand(3,1),]*5) 
plt.show() 

Результат: