def rename_headers()

in datasets/chicago_crime/pipelines/_images/run_csv_transform_kub/csv_transform.py [0:0]

• 26 lines of code
• 1 McCabe index (conditional complexity)

def rename_headers(df: pd.DataFrame) -> None:
    header_names = {
        "ID": "unique_key",
        "Case Number": "case_number",
        "Date": "date",
        "Block": "block",
        "IUCR": "iucr",
        "Primary Type": "primary_type",
        "Description": "description",
        "Location Description": "location_description",
        "Arrest": "arrest",
        "Domestic": "domestic",
        "Beat": "beat",
        "District": "district",
        "Ward": "ward",
        "Community Area": "community_area",
        "FBI Code": "fbi_code",
        "X Coordinate": "x_coordinate",
        "Y Coordinate": "y_coordinate",
        "Year": "year",
        "Updated On": "updated_on",
        "Latitude": "latitude",
        "Longitude": "longitude",
        "Location": "location",
    }

    df.rename(columns=header_names, inplace=True)