Skip to content
Snippets Groups Projects
Commit 6353335e authored by Rok Roškar's avatar Rok Roškar Committed by Rok Roškar
Browse files

chore: update converters

parent e873da25
No related branches found
No related tags found
1 merge request!139add-example-notebook
......@@ -103,6 +103,8 @@ class CaseConverterImpl:
except TypeError:
pass
# make sure that date is datetime
df["date"] = pd.to_datetime(df["date"])
return df[self.common_columns]
# def _compute_daily_counts(self, df):
......
......@@ -37,7 +37,6 @@ class ItalyRegionalCaseConverter(CaseConverter):
metadata["istatid"] = metadata.istatid.astype(int)
df_conv = _correct_trentino(df_conv)
merged = pd.merge(df_conv, metadata, on="istatid").drop_duplicates()
return self._set_common_columns(merged)
def read_data(self, path):
......
......@@ -72,6 +72,7 @@ class SpainCaseConverter(CaseConverter):
# calculate incidence rates
merged = df_conv.merge(pd.DataFrame(region_populations))
merged["country"] = "ESP"
merged["country_label"] = "Spain"
return self._set_common_columns(merged)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment