code: def ax(self):
          
          if not hasattr(self, "_ax") or self._ax is None:
              self._ax = plt.gca()
          return self._ax
docs: The matplotlib axes that the visualizer draws upon (can also be a grid
          of multiple axes objects). The visualizer automatically creates an
          axes for the user if one has not been specified.