diff options
Diffstat (limited to 'plotter/meteogram.py')
| -rwxr-xr-x | plotter/meteogram.py | 27 |
1 files changed, 22 insertions, 5 deletions
diff --git a/plotter/meteogram.py b/plotter/meteogram.py index ee8c7ea..790714e 100755 --- a/plotter/meteogram.py +++ b/plotter/meteogram.py @@ -2,6 +2,8 @@ import os import json +import numpy as np + import matplotlib.pyplot as plt import xarray as xr @@ -51,14 +53,29 @@ def _plot(data, output, name, lat, lon): ax.invert_yaxis() - ### Second plot - + ### Temp + Dewpoint ax2 = fig.add_subplot(5,1,3,sharex=ax) ax2.plot(data.valid_time, data.t2m.metpy.convert_units('degC').transpose(), color='red', label='Temperature (2m)') ax2.plot(data.valid_time, mpcalc.dewpoint_from_relative_humidity(data.t2m, data.r2).transpose(), color='blue', label='Dewpoint (2m)') ax2.set_ylabel('Temperature [degC]') ax2.legend(loc='lower right') + ## MSLP + ax3 = fig.add_subplot(5,1,4,sharex=ax) + ax3.plot(data.valid_time, data.prmsl.metpy.convert_units('hPa').transpose(), color='black', label='Temperature (2m)') + ax3.set_ylabel('Mean Sea Level Pressure [hPa]') + #ax3.legend(loc='lower right') + # TODO: ADD HBAS_CON, HTOP_CON + # If none: -500m + + ax4 = ax3.twinx() + ax4.set_ylim(0, 14) + ax4.set_ylabel('Convective Clouds Height [km]') + ax4.bar(data.valid_time, + bottom=data.HBAS_CON.metpy.convert_units('km').transpose(), + height=(data.hcct.metpy.convert_units('km')-data.HBAS_CON.metpy.convert_units('km')).transpose(), + align='edge', width=np.timedelta64(3, 'h')) + ### Info Lines info_lines = [] @@ -71,10 +88,10 @@ def _plot(data, output, name, lat, lon): if '_description' in data.attrs: info_lines.append(data.attrs['_description']) - ax = fig.add_subplot(5, 1, 4) - ax.text(0, 0, '\n'.join(info_lines), ha='left', va='center', + ax_text = fig.add_subplot(5, 1, 5) + ax_text.text(0, 0, '\n'.join(info_lines), ha='left', va='center', size=10, fontfamily='monospace') - ax.axis("off") + ax_text.axis("off") ### Output |