refactoring

2022-11-22 12:40:26 +01:00
parent 134a133da1
commit e87fe0c8ba
21 changed files with 692 additions and 731 deletions
--- a/src/DataFrameFilter.py
+++ b/src/DataFrameFilter.py
@@ -0,0 +1,9 @@
+import pandas as pd
+
+class DataFrameFilter:
+    
+    def filterByCovid19(self, dataFrame):
+        return dataFrame[self._isCovid19(dataFrame)]
+
+    def _isCovid19(self, dataFrame):
+        return dataFrame["VAX_TYPE"] == "COVID19"