class_docstr: Base class defining the distribution default behavior and interface
docstr: Gets the column(s) associated with the instantiated DistType.
  
  Args:
      X: Full dataset to be selected from.
  
  Returns:
      1d or 2d np.ndarray of columns.
docstr_tok: ["Gets","the","column","(","s",")","associated","with","the","instantiated","DistType",".","Args",":","X",":","Full","dataset","to","be","selected","from",".","Returns",":","1d","or","2d","np.ndarray","of","columns","."]
code: def get_columns(
      self,
      X: np.ndarray,
  ) -> np.ndarray:
      """
      Gets the column(s) associated with the instantiated DistType.
  
      Args:
          X: Full dataset to be selected from.
  
      Returns:
          1d or 2d np.ndarray of columns.
      """
      return X[:, self.idx]
  
imports: import itertools
  from abc import ABCMeta, abstractmethod
  from copy import deepcopy
  from typing import Dict, List, Tuple
  import numpy
  import torch
  from causalnex.structure.structuremodel import StructureModel
label_desc: Use image node_id 2 for calling the DistTypeBase obj's underlying member method code with example usage: obj.get_columns(X) and returns: X