Plot multiple variables

This example shows how to plot multiple variables from a GenieModel object. I use a modern model run as an example and plot the surface temperature, PO4, Fe, and O2.

The following features are particularly demonstrated:

1. Customizing the plot projection with cartopy

2. Plotting multiple variables in one figure

3. Controlling the elements in cgeniepy.GriddedDataVis object

4. Using CommunityPalette to get a pretty color palette

/home/docs/checkouts/readthedocs.org/user_builds/cgeniepy/envs/latest/lib/python3.12/site-packages/cgeniepy/model.py:58: UserWarning: No gemflag is provided, use default gemflags: [biogem]
  warnings.warn("No gemflag is provided, use default gemflags: [biogem]")

import matplotlib.pyplot as plt
import cartopy.crs as ccrs
import cgeniepy
from cgeniepy.plot import CommunityPalette

fig, axs = plt.subplots(2, 2, figsize=(10, 8), subplot_kw={'projection': ccrs.PlateCarree()})

pi_model = cgeniepy.sample_model()

variable = ['ocn_sur_temp', 'bio_fexport_POC', 'ocn_sur_PO4','ocn_sur_O2']
cmap = ['r_terrain', 'tol_rainbow', 'parula','ODV']

for i in range(4):
    data = pi_model.get_var(variable[i]).isel(time=-1).to_GriddedDataVis()
    data.aes_dict['pcolormesh_kwargs']['cmap'] = CommunityPalette(cmap[i]).colormap
    data.aes_dict['contourf_kwargs']['cmap'] = CommunityPalette(cmap[i]).colormap

    data.aes_dict['facecolor_kwargs']['c'] = 'white'
    data.plot(ax=axs.flatten()[i],colorbar=True, outline=True, gridline=True, contourf=True)