Отнесите возраст к другой возрастной группе столбца

Question

У меня есть этот набор данных

age
24
32
29
23
23
31
25
26
34

Я хочу классифицировать, используя python, и сохранить результат в новом столбце «agegroup», чтобы возраст между; 23–26, чтобы вернуть 1 в столбце возрастной группы, 27-30, чтобы вернуть значение 2 в столбце возрастной группы, и 31-34, чтобы вернуть 3 в столбце возрастной группы

Answer 1

Вы можете использовать pandas.cut.

Дано:

>>> df
   age
0   24
1   32
2   29
3   23
4   23
5   31
6   25
7   26
8   34

Решение:

>>> df.assign(agegroup=pd.cut(df['age'], bins=[23, 27, 31, 35], right=False, labels=[1, 2, 3]))
   age agegroup
0   24        1
1   32        3
2   29        2
3   23        1
4   23        1
5   31        3
6   25        1
7   26        1
8   34        3

Answer 2

Для этого также можно использовать словари. Пары ключ-значение. Ключи будут разными возрастными диапазонами, а значение для определенного ключа будет счетчиком для этой конкретной возрастной группы.

groupDict = {'23-26': 0, '27-30 ': 0, '31-34': 0}

for i in ages:
 if i>=23 and i<=26:
   groupDict['23-26']+=1
 elif i>=27 and i<=30:
   groupDict['27-30']+=1
 elif i>=31 and i<=34:
   groupDict['27-30']+=1