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

clustering: add docstring to cluster_regions function, fix network reference

Browse Source
This commit is contained in:
Fabian 2024-04-11 12:13:07 +02:00
parent 47134a88d8
commit 590d363579
2 changed files with 25 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
scripts/cluster_network.py
View File

@ -428,7 +428,20 @@ def clustering_for_n_clusters(
return clustering return clustering
def cluster_regions(busmaps, which, input=None, output=None): def cluster_regions(n, busmaps, which, input=None, output=None):
"""
Cluster regions based on busmaps and save the results to a file and to the
network.
Parameters:
- busmaps (list): A list of busmaps used for clustering.
- which (str): The type of regions to cluster.
- input (str, optional): The input file path. Defaults to None.
- output (str, optional): The output file path. Defaults to None.
Returns:
None
"""
busmap = reduce(lambda x, y: x.map(y), busmaps[1:], busmaps[0]) busmap = reduce(lambda x, y: x.map(y), busmaps[1:], busmaps[0])
regions = gpd.read_file(getattr(input, which)) regions = gpd.read_file(getattr(input, which))
@ -438,7 +451,7 @@ def cluster_regions(busmaps, which, input=None, output=None):
regions_c = regions_c.reset_index() regions_c = regions_c.reset_index()
regions_c.to_file(getattr(output, which)) regions_c.to_file(getattr(output, which))
# remove old regions # remove original regions
remove = n.shapes.query("component == 'Bus' and type == @which").index remove = n.shapes.query("component == 'Bus' and type == @which").index
n.mremove("Shape", remove) n.mremove("Shape", remove)
@ -456,8 +469,8 @@ def cluster_regions(busmaps, which, input=None, output=None):
) )
def plot_busmap_for_n_clusters(n, n_clusters, fn=None): def plot_busmap_for_n_clusters(n, n_clusters, solver_name="scip", fn=None):
busmap = busmap_for_n_clusters(n, n_clusters) busmap = busmap_for_n_clusters(n, n_clusters, solver_name)
cs = busmap.unique() cs = busmap.unique()
cr = sns.color_palette("hls", len(cs)) cr = sns.color_palette("hls", len(cs))
n.plot(bus_colors=busmap.map(dict(zip(cs, cr)))) n.plot(bus_colors=busmap.map(dict(zip(cs, cr))))
@ -554,17 +567,15 @@ if __name__ == "__main__":
params.focus_weights, params.focus_weights,
) )
update_p_nom_max(clustering.network) nc = clustering.network
update_p_nom_max(nc)
if params.cluster_network.get("consider_efficiency_classes"): if params.cluster_network.get("consider_efficiency_classes"):
labels = [f" {label} efficiency" for label in ["low", "medium", "high"]] labels = [f" {label} efficiency" for label in ["low", "medium", "high"]]
nc = clustering.network
nc.generators["carrier"] = nc.generators.carrier.replace(labels, "", regex=True) nc.generators["carrier"] = nc.generators.carrier.replace(labels, "", regex=True)
clustering.network.meta = dict( nc.meta = dict(snakemake.config, **dict(wildcards=dict(snakemake.wildcards)))
snakemake.config, **dict(wildcards=dict(snakemake.wildcards)) nc.export_to_netcdf(snakemake.output.network)
)
clustering.network.export_to_netcdf(snakemake.output.network)
for attr in ( for attr in (
"busmap", "busmap",
"linemap", "linemap",
@ -572,4 +583,6 @@ if __name__ == "__main__":
getattr(clustering, attr).to_csv(snakemake.output[attr]) getattr(clustering, attr).to_csv(snakemake.output[attr])
for which in ["regions_onshore", "regions_offshore"]: for which in ["regions_onshore", "regions_offshore"]:
cluster_regions((clustering.busmap,), which, snakemake.input, snakemake.output) cluster_regions(
nc, (clustering.busmap,), which, snakemake.input, snakemake.output
)

2
scripts/simplify_network.py
View File

@ -630,4 +630,4 @@ if __name__ == "__main__":
busmap_s.to_csv(snakemake.output.busmap) busmap_s.to_csv(snakemake.output.busmap)
for which in ["regions_onshore", "regions_offshore"]: for which in ["regions_onshore", "regions_offshore"]:
cluster_regions(busmaps, which, snakemake.input, snakemake.output) cluster_regions(n, busmaps, which, snakemake.input, snakemake.output)