Files
HowBadIsMyBatch/src/CountryCountsByClickedBatchcodeProvider.py

34 lines
1.1 KiB
Python

import pandas as pd
class CountryCountsByClickedBatchcodeProvider:
@staticmethod
def getCountryCountsByClickedBatchcode(file):
exploration = pd.read_csv(file, index_col = 0, skiprows = [0, 1, 2, 3, 4, 5, 7])
exploration.index.name = 'VAX_LOT'
exploration.rename(
columns =
{
'Country': 'COUNTRY',
'Event count': 'COUNTRY_COUNT_BY_VAX_LOT'
},
inplace = True)
exploration.set_index('COUNTRY', append = True, inplace = True)
return exploration
@staticmethod
def getCityCountsByClickedBatchcode(file):
exploration = pd.read_csv(file, index_col = 0, skiprows = [0, 1, 2, 3, 4, 5, 7])
exploration.index.name = 'VAX_LOT'
exploration.rename(
columns =
{
'Country': 'COUNTRY',
'Region': 'REGION',
'City': 'CITY',
'Event count': 'CITY_COUNT_BY_VAX_LOT'
},
inplace = True)
exploration.set_index(['COUNTRY', 'REGION', 'CITY'], append = True, inplace = True)
return exploration