code: def dispersion(words, corpus, y=None, ax=None, colors=None, colormap=None,
                 labels=None, annotate_docs=False, ignore_case=False, **kwargs):
      
  
      # Instantiate the visualizer
      visualizer = DispersionPlot(
          words, ax=ax, colors=colors, colormap=colormap,
          ignore_case=ignore_case, labels=labels,
          annotate_docs=annotate_docs, **kwargs
      )
  
      # Fit and transform the visualizer (calls draw)
      visualizer.fit(corpus, y, **kwargs)
  
      # Return the axes object on the visualizer
      return visualizer.ax
docs: Displays lexical dispersion plot for words in a corpus
  
      This helper function is a quick wrapper to utilize the DisperstionPlot
      Visualizer for one-off analysis
  
      Parameters
      ----------
  
      words : list
          A list of words whose dispersion will be examined within a corpus
  
      y : ndarray or Series of length n
          An optional array or series of target or class values for
          instances. If this is specified, then the points will be colored
          according to their class.
  
      corpus : list
          Should be provided as a list of documents that contain
          a list of words in the order they appear in the document.
  
      ax : matplotlib axes, default: None
          The axes to plot the figure on.
  
      labels : list of strings
          The names of the classes in the target, used to create a legend.
          Labels must match names of classes in sorted order.
  
      colors : list or tuple of colors
          Specify the colors for each individual class
  
      colormap : string or matplotlib cmap
          Qualitative colormap for discrete target
  
      annotate_docs : boolean, default: False
          Specify whether document boundaries will be displayed.  Vertical lines
          are positioned at the end of each document.
  
      ignore_case : boolean, default: False
  	Specify whether input  will be case-sensitive.
  
      kwargs : dict
          Pass any additional keyword arguments to the super class.
  
      Returns
      -------
      ax: matplotlib axes
          Returns the axes that the plot was drawn on