Категоризировать столбец на основе значений другого столбца

Question

У меня есть DataFrame следующим образом:
col1 num agg_col
12 200 0
13 300 0
14 400 0
15 500 0
16600 0
17 700 0

Я пытаюсь заполнить agg_col на основе значений в col1.
Например, если col1 равно 12–14, заполнить 1 в agg_col, 15-16,
заполнить 2 на agg_col.если col1 = 17, заполните 3.

Я написал следующий код Python:

df['agg_col'][(df['col1'] >= 12) & (df['col1'] <= 14)] = 1

Но я застрял здесь и не могу продолжить.Пожалуйста, помогите !!!

Answer 1

Попробуйте взглянуть на pd.cut

pd.cut(df.col1,[0,15,16,17],labels=[1,2,3])
Out[988]: 
0    1
1    1
2    1
3    1
4    2
5    3
Name: col1, dtype: category
Categories (3, int64): [1 < 2 < 3]