refactoring

2022-12-14 22:51:36 +01:00
parent 256ccd926d
commit 1e7e859f9c
2 changed files with 17 additions and 49 deletions
--- a/src/intensivstationen/IntensiveCareBedsProvider.py
+++ b/src/intensivstationen/IntensiveCareBedsProvider.py
@@ -0,0 +1,12 @@
+import pandas as pd
+
+
+def getIntensiveCareBeds(timeSeries, kreis = None):
+    if kreis is not None:
+        return timeSeries[timeSeries['Kreis'] == kreis][['date', 'betten_belegt', 'betten_frei', 'Einwohnerzahl']]
+    else:
+        return timeSeries.groupby('date').agg(**{
+                        'betten_belegt': pd.NamedAgg(column = 'betten_belegt', aggfunc = 'sum'),
+                        'betten_frei':   pd.NamedAgg(column = 'betten_frei',   aggfunc = 'sum'),
+                        'Einwohnerzahl': pd.NamedAgg(column = 'Einwohnerzahl', aggfunc = 'sum')
+                    }).reset_index()