def reorder_headers()

in datasets/census_bureau_international/pipelines/_images/run_csv_transform_kub_mortality_life_expectancy/csv_transform.py [0:0]


def reorder_headers(df: pd.DataFrame) -> pd.DataFrame:
    logging.info("Reordering headers..")
    df = df[
        [
            "country_code",
            "country_name",
            "year",
            "infant_mortality",
            "infant_mortality_male",
            "infant_mortality_female",
            "life_expectancy",
            "life_expectancy_male",
            "life_expectancy_female",
            "mortality_rate_under5",
            "mortality_rate_under5_male",
            "mortality_rate_under5_female",
            "mortality_rate_1to4",
            "mortality_rate_1to4_male",
            "mortality_rate_1to4_female",
        ]
    ]

    return df