original_docstring: Plot the heatmap for policy or values
  
      :param data:
      :param title:
      :param axes:
      :param iteration:
      :return:
      
docstring: Plot the heatmap for policy or values
docstring_tokens: Plot the heatmap for policy or values
code: def heatmap(data, title, axes, iteration):
      """Plot the heatmap for policy or values
  
      :param data:
      :param title:
      :param axes:
      :param iteration:
      :return:
      """
      h = sns.heatmap(data, ax=axes[iteration])
      h.set_ylabel('#Cars at first location')
      h.set_xlabel('#Cars at second location')
      h.set_title(title)
      h.invert_yaxis()
      plt.rc('mathtext', fontset="cm")
code_tokens: def heatmap ( data , title , axes , iteration ) : h = sns . heatmap ( data , ax = axes [ iteration ] ) h . set_ylabel ( '#Cars at first location' ) h . set_xlabel ( '#Cars at second location' ) h . set_title ( title ) h . invert_yaxis ( ) plt . rc ( 'mathtext' , fontset = "cm" )
short_docstring: Plot the heatmap for policy or values
short_docstring_tokens: Plot the heatmap for policy or values
comment: """Plot the heatmap for policy or values
  
      :param data:
      :param title:
      :param axes:
      :param iteration:
      :return:
      """