In [12]:

    

import matplotlib.pyplot as plt
import numpy as np


    

In [7]:

    

model_list = ['CNRM-CM6-1', 'CNRM-ESM2-1', 'GFDL-CM4', 'HadGEM3-GC31-LL', 'UKESM1-0-LL']

hfds_drift = {}
hfds_drift['CNRM-CM6-1'] = 1.80904332607144E+21
hfds_drift['CNRM-ESM2-1'] = 1.13000665464596E+21
hfds_drift['GFDL-CM4'] = 3.00264105055233E+21
hfds_drift['HadGEM3-GC31-LL'] = -1.52288909378555E+21
hfds_drift['UKESM1-0-LL'] = -4.03426986785994E+21

hfdsgeou_drift = {}
hfdsgeou_drift['CNRM-CM6-1'] = 2.55966936485025E+21
hfdsgeou_drift['CNRM-ESM2-1'] = 1.88063269342477E+21
hfdsgeou_drift['GFDL-CM4'] = 4.08107808572345E+21
hfdsgeou_drift['HadGEM3-GC31-LL'] = -782292856456320000000
hfdsgeou_drift['UKESM1-0-LL'] = -3.29367363053071E+21

ohct_drift = {}
ohct_drift['CNRM-CM6-1'] = 1.87263063520929E+21
ohct_drift['CNRM-ESM2-1'] = 1.20006589699253E+21
ohct_drift['GFDL-CM4'] = 3.23794878562647E+21
ohct_drift['HadGEM3-GC31-LL'] = 1.91149386234439E+21
ohct_drift['UKESM1-0-LL'] = -657982481269016000000

model_color = {}
model_color['CNRM-CM6-1'] = 'red'
model_color['CNRM-ESM2-1'] = 'gold'
model_color['GFDL-CM4'] = 'green'
model_color['HadGEM3-GC31-LL'] = 'blue'
model_color['UKESM1-0-LL'] = 'purple'


    

In [24]:

    

fig = plt.figure(figsize=[10, 7])
ax = fig.add_subplot(111)

for model in model_list:
    ax.scatter(hfdsgeou_drift[model], ohct_drift[model],
               color=model_color[model], label=f"{model}, hfds + hfgeou", marker='o')
    ax.scatter(hfds_drift[model], ohct_drift[model],
               color=model_color[model], label=f"{model}, hfds", marker='x')
ax.legend()
ax.axhline(y=0, color='black', linewidth=1.0)
ax.axvline(x=0, color='black', linewidth=1.0)

xlim = ax.get_xlim()
ylim = ax.get_ylim()
x_vals = np.array(xlim)
y_vals = x_vals
ax.plot(x_vals, y_vals, linestyle='--', c='0.5')

ax.set_title('(b) ocean energy conservation')
ax.set_xlabel('cumulative heat flux into ocean, $dQ_h/dt \: (J \: y^{-1})$')
ax.set_ylabel('change in OHC temperature component, $dH_T/dt \: (J \: y^{-1})$')

plt.show()


    

In [ ]: