在Python和Matplotlib中为3D绘图添加不同的阴影

Question

我正在寻找一种非常简单的方法来改变3D绘图的颜色，使其看起来更有趣。

这是我的代码（不得不删除大部分内容，因为我的帖子中有太多代码）：

from numpy import *
from matplotlib.pyplot import *
from mpl_toolkits.mplot3d import Axes3D
from numpy import array
from matplotlib import pyplot
from math import sqrt    

X, Y = np.mgrid[:9, :21]

fig = figure()
ax = fig.add_subplot(1,1,1, projection='3d')
ax.set_xlabel('Time')
ax.set_ylabel('Step')
ax.set_zlabel('Probability')

ax.plot_surface(X, Y, probs)
show()

这是我的图表：

真的很感激任何帮助!!

我尝试使用http://matplotlib.org/mpl_toolkits/mplot3d无效