From 567332695bcdab799f94e958bb7e60b945a29472 Mon Sep 17 00:00:00 2001 From: oussamanaji Date: Fri, 10 Apr 2020 17:14:28 +0300 Subject: [PATCH] Update fr_covidata.py --- task_geo/data_sources/covid/fr_covidata/fr_covidata.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/task_geo/data_sources/covid/fr_covidata/fr_covidata.py b/task_geo/data_sources/covid/fr_covidata/fr_covidata.py index 4ecbeb0..2b21c7d 100644 --- a/task_geo/data_sources/covid/fr_covidata/fr_covidata.py +++ b/task_geo/data_sources/covid/fr_covidata/fr_covidata.py @@ -67,11 +67,10 @@ def fr_covidata_formatter(dataset): frcovidata(pandas.DataFrame) """ - no_granularites = ['region', 'monde', 'pays', 'collectivite-outremer'] - no_maille_codes = ['DEP-971', 'DEP-972', 'DEP-973', 'DEP-974', 'DEP-976'] + no_gr = ['region', 'monde', 'pays', 'collectivite-outremer'] + no_mc = ['DEP-971', 'DEP-972', 'DEP-973', 'DEP-974', 'DEP-976'] dataset = dataset[ - (~dataset.granularite.isin(no_granularites)) & - (~dataset.maille_code.isin(no_maille_codes)) + (~dataset.granularite.isin(no_gr)) & (~dataset.maille_code.isin(no_mc)) ] dataset = dataset.drop(['depistes', 'granularite'], axis=1) dataset = dataset.drop_duplicates(