NVIDIA / earth2studio

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.
https://nvidia.github.io/earth2studio/
Apache License 2.0
73 stars 23 forks source link

Updating IMERG total precip data to meters #76

Closed NickGeneva closed 2 months ago

NickGeneva commented 2 months ago

Earth2Studio Pull Request

Description

IMERG provides total precip in mm, add modifier in lexicon to move to m. https://arthurhou.pps.eosdis.nasa.gov/Documents/IMERG_TechnicalDocumentation_final.pdf

import datetime
from earth2studio.data import IMERG, ARCO

date = datetime.datetime(year=2002, month=10, day=1)

wbds = ARCO(cache=False)
imergds = IMERG(cache=False)

wb_data = wbds(time=date, variable=["tp"])
im_data = imergds(time=date, variable=["tp"])

import matplotlib.pyplot as plt
from matplotlib.colors import LogNorm

fig, ax = plt.subplots(1, 2)

ax[0].imshow(wb_data.values[0, 0] + 0.001, norm=LogNorm(vmin=0.0001, vmax=0.1))
ax[0].set_title("ERA5 tp")
ax[1].imshow(im_data.values[0, 0] + 0.001, norm=LogNorm(vmin=0.0001, vmax=0.1))
ax[1].set_title("IMERG tp")

plt.savefig("test.png")

test

Checklist

Dependencies

NickGeneva commented 2 months ago

/blossom-ci