Plot the heat map for the selected variable.

Parameters¶

variable: str

The variable to plot

title: str, optional

The plot title

ax: pyplot.Axis, optional

The axis

figsize: tuple

The figsize argument for plt.subplots() in case ax is not provided

kwargs: dict, optional

Additional parameters for plt.pcolormesh()

Returns¶

ax: pyplot.Axis

The plot axis