我在一个类中有这个代码:
def graph(self, color = 'viridis'):
# Other colors: https://matplotlib.org/examples/color/colormaps_reference.html
fig = plt.figure()
Z = np.array(self.result).reshape(self.lenght_axes, self.lenght_axes)
ax = fig.add_subplot(1, 2, 1, projection='3d')
X = np.arange(self.lenght_axes)
Y = np.arange(self.lenght_axes)
X, Y = np.meshgrid(X, Y)
d3 = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=color, linewidth=0, antialiased=False)
fig.colorbar(d3)
ax = fig.add_subplot(1, 2, 2)
d2 = ax.imshow(Z, cmap=color, interpolation='none')
fig.colorbar(d2)
plt.show()
当我执行此操作时,我得到了这个:
我的问题是我如何配置彩条以使两者具有相同的数字?
如您所见,其中一个颜色条有0.6,0.4,0.2,0.0,-0.2,-0.4,-0.6,另一个有0.75,0.5,0.25,0.00,-0.25,-0.50,-0.75。但这很奇怪,因为两个图都是用相同的数据(self.result
)制作的,所以我不明白为什么这些条有不同的值。我想修改它以获得相同的数字。