23 lines
769 B
Python
23 lines
769 B
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import atlite
|
||
|
|
import pandas as pd
|
||
|
|
from vresutils import shapes as vshapes, hydro as vhydro
|
||
|
|
import logging
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
logger.setLevel(level=snakemake.config['logging_level'])
|
||
|
|
|
||
|
|
cutout = atlite.Cutout(snakemake.config['renewable']['hydro']['cutout'])
|
||
|
|
|
||
|
|
countries = snakemake.config['countries']
|
||
|
|
country_shapes = pd.Series(vshapes.countries(countries)).reindex(countries)
|
||
|
|
country_shapes.index.name = 'countries'
|
||
|
|
|
||
|
|
eia_stats = vhydro.get_eia_annual_hydro_generation().reindex(columns=countries)
|
||
|
|
inflow = cutout.runoff(shapes=country_shapes,
|
||
|
|
smooth=True,
|
||
|
|
lower_threshold_quantile=True,
|
||
|
|
normalize_using_yearly=eia_stats)
|
||
|
|
|
||
|
|
inflow.to_netcdf(snakemake.output[0])
|