sid.virus_strains

Module Contents

Functions

prepare_virus_strain_factors(virus_strains: Dict[str, List[str]], params: pd.DataFrame) → Dict[str, Union[List[str], np.ndarray]]	Prepare the information on virus strains and factors for infectiousness.
combine_first_factorized_infections(first: np.ndarray, second: np.ndarray) → np.ndarray	Combine factorized infections where the first has precedence.
categorize_factorized_infections(factorized_infections: Union[pd.Series, np.ndarray], virus_strains: Dict[str, Any]) → pd.Series	Convert factorized infections with virus strains to a categorical.
factorize_initial_infections(infections: pd.DataFrame, virus_strains: Dict[str, Any]) → pd.DataFrame	Factorize multiple boolean or categorical infections.
factorize_boolean_or_categorical_infections(infections, virus_strains)	Factorize boolean or categorical infections.
_factorize_boolean_infections(infected: Union[pd.Series, np.ndarray], names: List[str]) → Tuple[np.ndarray]	Factorize boolean infection.
factorize_categorical_infections(virus_strain: pd.Series, names: List[str]) → Tuple[np.ndarray]	Factorize a categorical variable indicating virus strains.

prepare_virus_strain_factors(virus_strains: Dict[str, List[str]], params: pd.DataFrame) → Dict[str, Union[List[str], np.ndarray]]

Prepare the information on virus strains and factors for infectiousness.

This function recreates the dictionary to not change the original value in partialed function and adds the factors.

combine_first_factorized_infections(first: np.ndarray, second: np.ndarray) → np.ndarray

Combine factorized infections where the first has precedence.

categorize_factorized_infections(factorized_infections: Union[pd.Series, np.ndarray], virus_strains: Dict[str, Any]) → pd.Series

Convert factorized infections with virus strains to a categorical.

factorize_initial_infections(infections: pd.DataFrame, virus_strains: Dict[str, Any]) → pd.DataFrame

Factorize multiple boolean or categorical infections.

factorize_boolean_or_categorical_infections(infections, virus_strains)

Factorize boolean or categorical infections.

_factorize_boolean_infections(infected: Union[pd.Series, np.ndarray], names: List[str]) → Tuple[np.ndarray]

Factorize boolean infection.

factorize_categorical_infections(virus_strain: pd.Series, names: List[str]) → Tuple[np.ndarray]

Factorize a categorical variable indicating virus strains.