From c543fb39c6201b153ec869d8018f70cddcb9a97e Mon Sep 17 00:00:00 2001 From: Fabian Neumann Date: Fri, 25 Sep 2020 12:34:34 +0200 Subject: [PATCH] cluster: fix removing deprec squeeze=True in groupby (#194) --- scripts/cluster_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/cluster_network.py b/scripts/cluster_network.py index ce1e4319..a4da7cda 100644 --- a/scripts/cluster_network.py +++ b/scripts/cluster_network.py @@ -258,7 +258,7 @@ def busmap_for_n_clusters(n, n_clusters, solver_name, focus_weights=None, algori raise ValueError(f"`algorithm` must be one of 'kmeans', 'spectral' or 'louvain'. Is {algorithm}.") return (n.buses.groupby(['country', 'sub_network'], group_keys=False) - .apply(busmap_for_country).rename('busmap')) + .apply(busmap_for_country).squeeze().rename('busmap')) def plot_busmap_for_n_clusters(n, n_clusters=50): busmap = busmap_for_n_clusters(n, n_clusters)