Files
HowBadIsMyBatch/src/InternationalVaersCovid19Provider.py
2023-01-28 10:15:59 +01:00

32 lines
1.3 KiB
Python

from DataFrameFilter import DataFrameFilter
import VaersReader
import pandas as pd
from VaersDescrReader import VaersDescrReader
def getInternationalVaersCovid19(years):
internationalVaers = pd.concat(
[
VaersReader.getVaersForYears(years),
VaersReader.getNonDomesticVaers()
])
internationalVaersCovid19 = DataFrameFilter().filterByCovid19(internationalVaers)
return internationalVaersCovid19
def get_international_VAERSVAX_VAERSSYMPTOMS_Covid19(years):
international_VAERSVAX, international_VAERSSYMPTOMS = _get_international_VAERSVAX_VAERSSYMPTOMS(years)
international_VAERSVAX.dropna(subset = ['VAX_LOT'], inplace = True)
international_VAERSVAX_Covid19 = DataFrameFilter().filterByCovid19(international_VAERSVAX)
return international_VAERSVAX_Covid19, international_VAERSSYMPTOMS
def _get_international_VAERSVAX_VAERSSYMPTOMS(years):
vaersDescrReader = VaersDescrReader(dataDir = "VAERS")
internationalVaersDescrs = vaersDescrReader.readVaersDescrsForYears(years) + [vaersDescrReader.readNonDomesticVaersDescr()]
return _getVaersDescrByName(internationalVaersDescrs, 'VAERSVAX'), _getVaersDescrByName(internationalVaersDescrs, 'VAERSSYMPTOMS')
def _getVaersDescrByName(vaersDescrs, vaersDescrName):
return pd.concat([vaersDescr[vaersDescrName] for vaersDescr in vaersDescrs])