creating a new column in a dataframe for an age class using the values in the age column of same dataframe
def my_categ(data): age_groups = [f”{i}-{i+4}” for i in range(0, 100, 5)] grp_data = {} for i in range(len(age_groups)): start_age = i * 5 end_age = start_age + 4 grp_data[age_groups[i]] = data.iloc[:, start_age:end_age+1].sum(axis=1) return grouped_data df[‘Age Class’] = df[‘Age’].apply(my_categ) df how do I fix this? python pandas dataframe function apply