poliosim.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

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)