20 lines
1.2 KiB
Python
20 lines
1.2 KiB
Python
from HistogramFactory import createGlobalHistograms, 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')
|
|
|
|
|
|
def createAndSaveHistogramsForCountries(symptomByBatchcodeTable):
|
|
dictByBatchcodeTable = createHistograms(symptomByBatchcodeTable)
|
|
explodedTable = MultiIndexExploder.explodeMultiIndexOfTable(dictByBatchcodeTable)
|
|
histogramDescriptionTable = HistogramDescriptionTableFactory.createHistogramDescriptionTable(explodedTable)
|
|
for country, histogramDescriptionTableForCountry in histogramDescriptionTable.groupby('COUNTRY'):
|
|
print(country)
|
|
saveHistograms(histogramDescriptionTableForCountry, country) |