- def categorize_income(row):
- total_income = row['total_income']
- if 0 <= total_income <= 30000:
- return 'E'
- if 30001 <= total_income <= 50000:
- return 'D'
- if 50001 <= total_income <= 200000:
- return 'C'
- if 200001 <= total_income <= 1000000:
- return 'B'
- if 1000001 <= total_income:
- return 'A'
- row_values = ['E']
- row_columns = ['total_income']
- row = pd.Series(data=row_values, index=row_columns)
- data['total_income_category'] = data.apply(categorize_income, axis=1)