environmentaltools.graphics.heatmap

environmentaltools.graphics.heatmap(data: ndarray, param: dict, cmap: str = 'bwr_r', type_: str = None, file_name: str = None, ax=None, minmax=False)[source]

Create a heatmap from a numpy array and two lists of labels.

Parameters:
  • data (*) – A 2D array of shape (N, M).

  • param (*) – A list or array of length N with the labels for the rows.

  • type (*) – type of variable to be plotted. B stands for the parameter matrix and Q for the covariance matrix

  • file_name (*) – name of the oputput file