pycontrails.datalib.ecmwf.ERA5¶

class pycontrails.datalib.ecmwf.ERA5(time, variables, pressure_levels=-1, paths=None, timestep_freq=None, product_type='reanalysis', grid=None, cachestore=<object object>, url=None, key=None)¶

Bases: ECMWFAPI

Class to support ERA5 data access, download, and organization.

Requires account with Copernicus Data Portal and local credentials.

API credentials can be stored in a ~/.cdsapirc file or as CDSAPI_URL and CDSAPI_KEY environment variables.

export CDSAPI_URL=… export CDSAPI_KEY=…

Credentials can also be provided directly url and key keyword args.

See cdsapi documentation for more information.

Parameters:

time (datalib.TimeInput | None) – The time range for data retrieval, either a single datetime or (start, end) datetime range. Input must be datetime-like or tuple of datetime-like (datetime, pd.Timestamp, np.datetime64) specifying the (start, end) of the date range, inclusive. Datafiles will be downloaded from CDS for each day to reduce requests. If None, paths must be defined and all time coordinates will be loaded from files.
variables (datalib.VariableInput) – Variable name (i.e. “t”, “air_temperature”, [“air_temperature, relative_humidity”])
pressure_levels (datalib.PressureLevelInput, optional) – Pressure levels for data, in hPa (mbar) Set to -1 for to download surface level parameters. Defaults to -1.
paths (str | list[str] | pathlib.Path | list[pathlib.Path] | None, optional) – Path to CDS NetCDF files to load manually. Can include glob patterns to load specific files. Defaults to None, which looks for files in the cachestore or CDS.
timestep_freq (str, optional) – Manually set the timestep interval within the bounds defined by time. Supports any string that can be passed to pd.date_range(freq=…). By default, this is set to “1h” for reanalysis products and “3h” for ensemble products.
product_type (str, optional) – Product type, one of “reanalysis”, “ensemble_mean”, “ensemble_members”, “ensemble_spread”
grid (float, optional) – Specify latitude/longitude grid spacing in data. By default, this is set to 0.25 for reanalysis products and 0.5 for ensemble products.
cachestore (cache.CacheStore | None, optional) – Cache data store for staging ECMWF ERA5 files. Defaults to cache.DiskCacheStore. If None, cache is turned off.
url (str) – Override cdsapi url
key (str) – Override cdsapi key

Notes

ERA5 parameter list: https://confluence.ecmwf.int/pages/viewpage.action?pageId=82870405#ERA5:datadocumentation-Parameterlistings

All radiative quantities are accumulated. See https://www.ecmwf.int/sites/default/files/elibrary/2015/18490-radiation-quantities-ecmwf-model-and-mars.pdf for more information.

Local paths are loaded using xarray.open_mfdataset(). Pass xr_kwargs inputs to open_metdataset() to customize file loading.

Examples

>>> from datetime import datetime
>>> from pycontrails.datalib.ecmwf import ERA5
>>> from pycontrails import GCPCacheStore

>>> # Store data files from CDS to local disk (default behavior)
>>> era5 = ERA5(
...     "2020-06-01 12:00:00",
...     variables=["air_temperature", "relative_humidity"],
...     pressure_levels=[350, 300]
... )

>>> # cache files to google cloud storage
>>> gcp_cache = GCPCacheStore(
...     bucket="contrails-301217-unit-test",
...     cache_dir="ecmwf",
... )
>>> era5 = ERA5(
...     "2020-06-01 12:00:00",
...     variables=["air_temperature", "relative_humidity"],
...     pressure_levels=[350, 300],
...     cachestore=gcp_cache
... )

__init__(time, variables, pressure_levels=-1, paths=None, timestep_freq=None, product_type='reanalysis', grid=None, cachestore=<object object>, url=None, key=None)¶

Methods

`__init__`(time, variables[, pressure_levels, ...])
`cache_dataset`(dataset)	Cache data from data source.
`create_cachepath`(t)	Return cachepath to local ERA5 data file based on datetime.
`download`(**xr_kwargs)	Confirm all data files are downloaded and available locally in the `cachestore`.
`download_dataset`(times)	Download data from data source for input times.
`is_datafile_cached`(t, **xr_kwargs)	Check datafile defined by datetime for variables and pressure levels in class.
`list_timesteps_cached`(**xr_kwargs)	Get a list of data files available locally in the `cachestore`.
`list_timesteps_not_cached`(**xr_kwargs)	Get a list of data files not available locally in the `cachestore`.
`open_dataset`(disk_paths, **xr_kwargs)	Open multi-file dataset in xarray.
`open_metdataset`([dataset, xr_kwargs])	Open MetDataset from data source.
`set_metadata`(ds)	Set met source metadata on `ds.attrs`.

Attributes

`product_type`	Product type, one of "reanalysis", "ensemble_mean", "ensemble_members", "ensemble_spread"
`cds`	Handle to `cdsapi.Client`
`url`	User provided `cdsapi.Client` url
`key`	User provided `cdsapi.Client` url
`dataset`	Select dataset for download based on `pressure_levels`.
`grid`	Lat / Lon grid spacing
`hash`	Generate a unique hash for this datasource.
`is_single_level`	Return True if the datasource is single level data.
`paths`	Path to local source files to load.
`pressure_level_variables`	ECMWF pressure level parameters.
`pressure_levels`	List of pressure levels.
`single_level_variables`	ECMWF surface level parameters.
`supported_pressure_levels`	Get pressure levels available from ERA5 pressure level dataset.
`supported_variables`	Parameters available from data source.
`timesteps`	List of individual timesteps from data source derived from `time` Use `parse_time()` to handle `TimeInput`.
`variable_ecmwfids`	Return a list of variable ecmwf_ids.
`variable_shortnames`	Return a list of variable short names.
`variable_standardnames`	Return a list of variable standard names.
`variables`	Variables requested from data source Use `parse_variables()` to handle `VariableInput`.
`cachestore`	Cache store for intermediates while processing data source If None, cache is turned off.

cds¶: Handle to cdsapi.Client

create_cachepath(t)¶

Return cachepath to local ERA5 data file based on datetime.

This uniquely defines a cached data file ith class parameters.

Parameters:: t (datetime | pd.Timestamp) – Datetime of datafile
Returns:: str – Path to local ERA5 data file

property dataset¶

Select dataset for download based on pressure_levels.

One of “reanalysis-era5-pressure-levels” or “reanalysis-era5-single-levels”

Returns:: str – ERA5 dataset name in CDS

download_dataset(times)¶

Download data from data source for input times.

Parameters:: times (list[datetime]) – List of datetimes to download a store in cache

property hash¶

Generate a unique hash for this datasource.

Returns:: str – Unique hash for met instance (sha1)

key¶: User provided cdsapi.Client url

open_metdataset(dataset=None, xr_kwargs=None, **kwargs)¶

Open MetDataset from data source.

This method should download / load any required datafiles and returns a MetDataset of the multi-file dataset opened by xarray.

Parameters:

dataset (xr.Dataset | None, optional) – Input xr.Dataset loaded manually. The dataset must have the same format as the original data source API or files.
xr_kwargs (dict[str, Any] | None, optional) – Dictionary of keyword arguments passed into xarray.open_mfdataset() when opening files. Examples include “chunks”, “engine”, “parallel”, etc. Ignored if dataset is input.
**kwargs (Any) – Keyword arguments passed through directly into MetDataset constructor.

Returns:

MetDataset – Meteorology dataset