pihnn.graphics.single_plot#

pihnn.graphics.single_plot(triangulation, z, levels, title=None, cmap='jet', **kwargs)#

Single contour plot. Used by the other functions in this module.

Parameters:
  • triangulation (matplotlib.tri.Triangulation) – 2D mesh used for the contour plot.

  • z (numpy.ndarray/torch.tensor) – Function to be plotted.

  • levels (numpy.ndarray) – Levels corresponding to the contour coloring.

  • title (str) – Title of the plot, None if no title is desired.

  • cmap (str) – Predefined matplotlib color map.