otld.utils.caseload_utils.clean_dataset

otld.utils.caseload_utils.clean_dataset(df: DataFrame) → DataFrame

Clean dataset

This function performs the following actions:

• Clean up State variable

• Remove rows not containing a state name

• If “U.S. Total” is not present, generate it by summing all other rows.

Args:

df (pd.DataFrame): Data frame to clean

Returns:

pd.DataFrame: Cleaned data frame