add start of state data transformation

datasets/transform_SIR.py

@@ -3,14 +3,7 @@ import pandas as pd
 
 from src.plotter import Plotter
 
-dataset_path = 'datasets/COVID-19-Todesfaelle_in_Deutschland/'
-
-FONT_COLOR = '#595959'
-SUSCEPTIBLE = '#6399f7'
-INFECTIOUS = '#f56262'
-REMOVED = '#83eb5e'
-
-def transform_general_data(plotter:Plotter, plot_name='', plot_title='', sample_rate=1, exclude=[], plot_size=(12,6), yscale_log=False, plot_legend=True):
+def transform_general_to_SIR(plotter:Plotter, dataset_path='datasets/COVID-19-Todesfaelle_in_Deutschland/', plot_name='', plot_title='', sample_rate=1, exclude=[], plot_size=(12,6), yscale_log=False, plot_legend=True):
     """Function to generate the SIR split from the data in the COVID-19-Todesfaelle_in_Deutschland dataset.
     """
     # read the data
@@ -75,4 +68,6 @@ def transform_general_data(plotter:Plotter, plot_name='', plot_title='', sample_
                              R[0::sample_rate]]) 
 
     np.savetxt(f"datasets/SIR_RKI_{sample_rate}.csv", COVID_Data, delimiter=",")
-     
+     
+def state_based_data(dataset_path='datasets/state_data/Aktuell_Deutschland_SarsCov2_Infektionen.csv/'):
+    pass