彩条值未与冲浪图映射

Question

我正在尝试在3d曲面图中绘制计数值，但是当我在冲浪图的代码中放置颜色条时，它在颜色条中显示了不同的值，因为数据在图中具有不同的值。帮我根据曲面图中的数据校正颜色条设置。

我的代码如下

''' 绘制conunt数据

'''

import numpy as np
import matplotlib.pyplot as plt
from pylab import genfromtxt;

from matplotlib.collections import LineCollection
from mpl_toolkits.mplot3d import Axes3D
from matplotlib.collections import PolyCollection
from matplotlib.colors import colorConverter
from matplotlib import cm

values=np.random.randint(0,5,3940)
count=np.reshape(values,(788,5))
fig1=plt.figure()
fig1.add_subplot(111)
plt.pcolor(count)
plt.colorbar()
plt.show()
r1=count.shape[0] # row data  unique no
c1=count.shape[1] # column data time value


print "r1=" , r1, "c1 =" ,c1

time=np.arange(0,5,1) # time data ( in sec)
unique=np.arange(0,r1,1) # energy information
fig2=plt.figure()
ax = fig2.gca(projection='3d')

X,Y=np.meshgrid(((time.T)*3e4)/1e6,2.04*unique)
surf = ax.plot_surface(X,Y,count, 
cmap=cm.coolwarm,linewidth=0, antialiased=False)
plt.colorbar(surf)
plt.show()