颜色条附加到图

Question

data = np.array(number_list) 
self.fig = plt.Figure() 
self.ax = self.fig.add_subplot(1,1,1) 


N = data.max() + 5 
self.ax.set_xlim(0,N) 
self.ax.set_ylim(0,N) 
self.ax.invert_yaxis() 

self.ax.get_xaxis().set_visible(False) 
self.ax.get_yaxis().set_visible(False) 
self.ax.set_aspect('equal') 
this = self.ax.scatter(data[:,1], data[:,0],color= 'black', marker = 's', s=3) 

self.fig.colorbar(this, ticks=[-1, 0, 1],orientation='horizontal') 

self.canvas = FigureCanvas(self, -1, self.fig) 
self.toolbar = NavigationToolBar(self.canvas) 
    self.vbox = wx.BoxSizer(wx.VERTICAL) 
    self.vbox.Add(self.canvas, 1, wx.LEFT | wx.TOP | wx.GROW) 
self.vbox.Add(self.toolbar,0,wx.EXPAND) 
    self.vbox.AddSpacer(25) 
    self.SetSizer(self.vbox) 
    self.vbox.Fit(self) 

我曾尝试创建一个“附加”颜色条，但是我得到错误TypeError: You must first set_array for mappable。该图是一个矩阵图。 我在哪里出错？

Answer 1

this = self.ax.scatter(data[:,1], data[:,0],color= 'black', marker = 's', s=3) 

self.fig.colorbar(this, ticks=[-1, 0, 1],orientation='horizontal') 

问题出现在这些行中。您已将所有标记设置为固定颜色，因此颜色条没有任何意义。如果你想使用colorbar你需要通过映射标量来设置标记的颜色 - >颜色，例如：

this = self.ax.scatter(data[:,1], data[:,0],color=color_data, marker = 's', s=3) 

self.fig.colorbar(this, ticks=[-1, 0, 1],orientation='horizontal') 

其中len(color_data) == len(data[:,0]，是一维数组状物体。