poliosim.data.loaders module

Load data

get_country_aliases()[source]

Define aliases for countries with odd names in the data

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

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)

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)