def categorize_income(total_income):

    if total_income <= 30000:

        return 'E'

    if total_income <= 50000:

        return 'D'

    if total_income <= 200000:

        return 'C'

    if total_income <= 1000000:

        return 'B'

    if total_income >= 1000001:

        return 'A'

    # создайте функцию categorize_income()

 

data['total_income_category'] = data['total_income'].apply(categorize_income)  # примените функцию методом apply()