code: def _axis(self, axis):
          
          return self.df.columns if axis == 0 else self.df.index
docs: Return the corresponding labels taking into account the axis.
  
          The axis could be horizontal (0) or vertical (1).