covasim.data.loaders module

Load data

get_country_aliases()

Define aliases for countries with odd names in the data

map_entries(json, location)

Find a match between the JSON file and the provided location(s).

Parameters:

• json (list or dict) – the data being loaded
• location (list or str) – the list of locations to pull from

show_locations(location=None, output=False)

Print a list of available locations.

Parameters:

• location (str) – if provided, only check if this location is in the list
• output (bool) – whether to return the list (else print)

Examples:

cv.data.show_locations() # Print a list of valid locations
cv.data.show_locations('lithuania') # Check if Lithuania is a valid location
cv.data.show_locations('Viet-Nam') # Check if Viet-Nam is a valid location

get_age_distribution(location=None)

Load age distribution for a given country or countries.

Parameters:location (str or list) – name of the country or countries to load the age distribution for Returns:Numpy array of age distributions, or dict if multiple locations Return type:age_data (array)

get_household_size(location=None)

Load household size distribution for a given country or countries.

Parameters:location (str or list) – name of the country or countries to load the age distribution for Returns:Size of household, or dict if multiple locations Return type:house_size (float)