bjoern/pypsa-eur
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

make_summary: remove snakemake dependencies

Browse Source
This commit is contained in:
martacki 2022-01-14 11:29:01 +01:00
parent 7067d924cc
commit 8cb5bd27d5
1 changed files with 17 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
scripts/make_summary.py
View File

@ -54,7 +54,7 @@ Replacing '/summaries/' with '/plots/' creates nice colored maps of the results.
""" """
import logging import logging
from _helpers import configure_logging from _helpers import configure_logging, retrieve_snakemake_keys
import os import os
import pypsa import pypsa
@ -378,7 +378,7 @@ outputs = ["costs",
] ]
def make_summaries(networks_dict, country='all'): def make_summaries(networks_dict, paths, config, country='all'):
columns = pd.MultiIndex.from_tuples(networks_dict.keys(),names=["simpl","clusters","ll","opts"]) columns = pd.MultiIndex.from_tuples(networks_dict.keys(),names=["simpl","clusters","ll","opts"])
@ -403,8 +403,8 @@ def make_summaries(networks_dict, country='all'):
n = n[n.buses.country == country] n = n[n.buses.country == country]
Nyears = n.snapshot_weightings.objective.sum() / 8760. Nyears = n.snapshot_weightings.objective.sum() / 8760.
costs = load_costs(tech_costs = snakemake.input[0], config = snakemake.config['costs'], costs = load_costs(tech_costs = paths[0], config = config['costs'],
elec_config = snakemake.config['electricity'], Nyears = Nyears) elec_config = config['electricity'], Nyears)
update_transmission_costs(n, costs, simple_hvdc_costs=False) update_transmission_costs(n, costs, simple_hvdc_costs=False)
assign_carriers(n) assign_carriers(n)
@ -431,25 +431,27 @@ if __name__ == "__main__":
network_dir = os.path.join('results', 'networks') network_dir = os.path.join('results', 'networks')
configure_logging(snakemake) configure_logging(snakemake)
def expand_from_wildcard(key): paths, config, wildcards, logs, out = retrieve_snakemake_keys(snakemake)
w = getattr(snakemake.wildcards, key)
return snakemake.config["scenario"][key] if w == "all" else [w]
if snakemake.wildcards.ll.endswith("all"): def expand_from_wildcard(key, config):
ll = snakemake.config["scenario"]["ll"] w = getattr(wildcards, key)
if len(snakemake.wildcards.ll) == 4: return config["scenario"][key] if w == "all" else [w]
ll = [l for l in ll if l[0] == snakemake.wildcards.ll[0]]
if wildcards.ll.endswith("all"):
ll = config["scenario"]["ll"]
if len(wildcards.ll) == 4:
ll = [l for l in ll if l[0] == wildcards.ll[0]]
else: else:
ll = [snakemake.wildcards.ll] ll = [wildcards.ll]
networks_dict = {(simpl,clusters,l,opts) : networks_dict = {(simpl,clusters,l,opts) :
os.path.join(network_dir, f'elec_s{simpl}_' os.path.join(network_dir, f'elec_s{simpl}_'
f'{clusters}_ec_l{l}_{opts}.nc') f'{clusters}_ec_l{l}_{opts}.nc')
for simpl in expand_from_wildcard("simpl") for simpl in expand_from_wildcard("simpl", config)
for clusters in expand_from_wildcard("clusters") for clusters in expand_from_wildcard("clusters")
for l in ll for l in ll
for opts in expand_from_wildcard("opts")} for opts in expand_from_wildcard("opts")}
dfs = make_summaries(networks_dict, country=snakemake.wildcards.country) dfs = make_summaries(networks_dict, paths, config, country=wildcards.country)
to_csv(dfs, snakemake.output[0]) to_csv(dfs, out[0])