code: def get_column_names(self, X):
          
          if isinstance(X, pd.DataFrame):
              return X.columns
  
          return range(X.shape[1])
docs: Return iterable containing columns for the given array X.
  
          Args:
              X: `numpy.ndarray` or `pandas.DataFrame`.
  
          Returns:
              iterable: columns for the given matrix.