从matplotlib中的图中删除颜色条

Question

这应该很容易，但是我很难过。基本上，我在matplotlib中有一个子图，每次调用一个函数时，我都会绘制一个六边形图，但是每次调用该函数时，都会得到一个新的颜色条，所以我真正想做的就是更新颜色条。不幸的是，这似乎不起作用，因为通过subplot.hexbin重新创建了附加了颜色条的对象。

def foo(self):
   self.subplot.clear()
   hb = self.subplot.hexbin(...)
   if self.cb:
      self.cb.update_bruteforce() # Doesn't work (hb is new)
   else:
      self.cb = self.figure.colorbar(hb)

我现在在这个烦人的地方，尝试完全删除颜色条轴，然后重新创建它。不幸的是，当我删除颜色条轴时，子图轴不会回收空间，并且调用self.subplot.reset_position（）并没有达到我的预期。

def foo(self):
   self.subplot.clear()
   hb = self.subplot.hexbin(...)
   if self.cb:
      self.figure.delaxes(self.figure.axes[1])
      del self.cb
      # TODO: resize self.subplot so it fills the 
      #    whole figure before adding the new colorbar
   self.cb = self.figure.colorbar(hb)

有没有人有什么建议？

非常感激！亚当

Answer 1

好吧，这是我的解决方案。不是很优雅，但也不是一个可怕的骇客。

def foo(self):
   self.subplot.clear()
   hb = self.subplot.hexbin(...)
   if self.cb:
      self.figure.delaxes(self.figure.axes[1])
      self.figure.subplots_adjust(right=0.90)  #default right padding
   self.cb = self.figure.colorbar(hb)

因为我只有一个子图，所以这可以满足我的需求。使用多个子图或在不同位置绘制颜色条时遇到相同问题的人将需要进行调整。