francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
frankknoll
2023-02-16 15:52:34 +01:00
parent 03ba89bcd5
commit 54ddc3e47f
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/HistogramFactoryAndPersister.py
View File

@@ -1,14 +1,12 @@
from HistogramFactory import createGlobalHistograms, createHistograms
from HistogramFactory import createHistograms
from HistogramPersister import saveHistograms
from MultiIndexExploder import MultiIndexExploder
from HistogramDescriptionTableFactory import HistogramDescriptionTableFactory
def createAndSaveGlobalHistograms(symptomByBatchcodeTable):
dictByBatchcodeTable4Country = createGlobalHistograms(symptomByBatchcodeTable)
explodedTable = MultiIndexExploder.explodeMultiIndexOfTable(dictByBatchcodeTable4Country)
histogramDescriptionTable = HistogramDescriptionTableFactory.createHistogramDescriptionTable(explodedTable)
saveHistograms(histogramDescriptionTable, 'Global')
symptomByBatchcodeTable = symptomByBatchcodeTable.assign(COUNTRY = 'Global')
createAndSaveHistogramsForCountries(symptomByBatchcodeTable)
def createAndSaveHistogramsForCountries(symptomByBatchcodeTable):
@@ -16,5 +14,5 @@ def createAndSaveHistogramsForCountries(symptomByBatchcodeTable):
explodedTable = MultiIndexExploder.explodeMultiIndexOfTable(dictByBatchcodeTable)
histogramDescriptionTable = HistogramDescriptionTableFactory.createHistogramDescriptionTable(explodedTable)
for country, histogramDescriptionTableForCountry in histogramDescriptionTable.groupby('COUNTRY'):
print(country)
print(f'saving histograms for {country}')
saveHistograms(histogramDescriptionTableForCountry, country)