From b90f83a659105ee49316027be4dc6c1d09e27c17 Mon Sep 17 00:00:00 2001 From: lisazeyen <35347358+lisazeyen@users.noreply.github.com> Date: Wed, 11 Sep 2024 09:35:31 +0200 Subject: [PATCH] simplify drop expression Co-authored-by: Fabian Neumann --- scripts/build_energy_totals.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/scripts/build_energy_totals.py b/scripts/build_energy_totals.py index 3d37a8ee..4f096262 100644 --- a/scripts/build_energy_totals.py +++ b/scripts/build_energy_totals.py @@ -698,11 +698,9 @@ def build_energy_totals( [countries, eurostat_years], names=["country", "year"] ) - to_drop = idees.columns[ - idees.columns.str.contains("space efficiency") - ^ idees.columns.str.contains("water efficiency") - ] - to_drop = to_drop.append(pd.Index(["passenger cars", "passenger car efficiency"])) + efficiency_keywords = ["space efficiency", "water efficiency"] + to_drop = idees.columns[~idees.columns.str.contains('|'.join(efficiency_keywords))] + to_drop = to_drop.append(["passenger cars", "passenger car efficiency"]) df = idees.reindex(new_index).drop(to_drop, axis=1)