Я пытаюсь изобразить свое значение счетчика на трехмерном графике, но когда я вставляю цветную полосу в код для диаграммы прибоя, она показывает другое значение на цветовой шкале, где данные имеют различное значение на графике. Помогите мне исправить настройку цветовой шкалы в соответствии с данными на графике поверхности.
Мой код следующий:
''»
построение константных данных
'' '
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()