Plot 2D transect of tracers in each basin

This example plots the modelled oxygen distribution in cGENIE.

The following features in the package are used:

1. Access data through cgeniepy.model module

2. A basin-mask operation

3. A linear interpolation

4. Get pretty color palette

5. Customise the plotting details

/home/docs/checkouts/readthedocs.org/user_builds/cgeniepy/envs/stable/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 cgeniepy
from cgeniepy.plot import CommunityPalette
import matplotlib.pyplot as plt

model = cgeniepy.sample_model()

fig, axs=plt.subplots(nrows=3, ncols=1, figsize=(6,9), tight_layout=True)

basins = ['Atlantic', 'Pacific', 'Indian']

cmap = CommunityPalette('tol_rainbow').colormap

for i in range(3):
    basin_data = model.get_var('ocn_O2').isel(time=-1).mask_basin(base='worjh2',basin=basins[i], subbasin='')
    basin_data_interp = basin_data.mean(dim='lon').interpolate(grid_number=50).to_GriddedDataVis()
    basin_data_interp.aes_dict['pcolormesh_kwargs']['cmap'] = cmap

    basin_data_interp.plot(ax=axs[i], contour=False, outline=True)
    axs[i].title.set_text(basins[i])

plt.show()