rsvsim.data.loaders module

Load data

rsvsim.data.loaders.get_country_aliases()[source]

Define aliases for countries with odd names in the data

rsvsim.data.loaders.map_entries(json, location)[source]

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

rsvsim.data.loaders.show_locations(location=None, output=False)[source]

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
rsvsim.data.loaders.get_age_distribution(location=None)[source]

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)

rsvsim.data.loaders.get_household_size(location=None)[source]

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)