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

cluster_network: Fix busmap_for_n_clusters for single country (fixes #18)

Browse Source 
Also update debugging helper.
This commit is contained in:
Jonas Hoersch 2019-02-20 21:08:36 +01:00
parent 9c78d6591d
commit 8b819569bc
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/cluster_network.py
View File

@ -125,7 +125,8 @@ def busmap_for_n_clusters(n, n_clusters, solver_name, algorithm="kmeans", **algo
else: else:
raise ArgumentError("`algorithm` must be one of 'kmeans', 'spectral' or 'louvain'") raise ArgumentError("`algorithm` must be one of 'kmeans', 'spectral' or 'louvain'")
return n.buses.groupby(['country', 'sub_network'], group_keys=False).apply(busmap_for_country) return (n.buses.groupby(['country', 'sub_network'], group_keys=False, squeeze=True)
.apply(busmap_for_country).rename('busmap'))
def plot_busmap_for_n_clusters(n, n_clusters=50): def plot_busmap_for_n_clusters(n, n_clusters=50):
busmap = busmap_for_n_clusters(n, n_clusters) busmap = busmap_for_n_clusters(n, n_clusters)
@ -187,12 +188,14 @@ if __name__ == "__main__":
input=Dict( input=Dict(
network='networks/{network}_s{simpl}.nc', network='networks/{network}_s{simpl}.nc',
regions_onshore='resources/regions_onshore_{network}_s{simpl}.geojson', regions_onshore='resources/regions_onshore_{network}_s{simpl}.geojson',
regions_offshore='resources/regions_offshore_{network}_s{simpl}.geojson' regions_offshore='resources/regions_offshore_{network}_s{simpl}.geojson',
clustermaps='resources/clustermaps_{network}_s{simpl}.h5'
), ),
output=Dict( output=Dict(
network='networks/{network}_s{simpl}_{clusters}.nc', network='networks/{network}_s{simpl}_{clusters}.nc',
regions_onshore='resources/regions_onshore_{network}_s{simpl}_{clusters}.geojson', regions_onshore='resources/regions_onshore_{network}_s{simpl}_{clusters}.geojson',
regions_offshore='resources/regions_offshore_{network}_s{simpl}_{clusters}.geojson' regions_offshore='resources/regions_offshore_{network}_s{simpl}_{clusters}.geojson',
clustermaps='resources/clustermaps_{network}_s{simpl}_{clusters}.h5'
) )
) )