Внесение результата работы функции в таблицу с помощью метода .apply ()

1) создаем функцию для разделения на категории
(она возвращает название категории в зависимости от соответствия условию)
(*отступы могут отображаться некорректно)

def my_function_name(smth_to_enter_in_function):

--------if smth_to_enter_in_function > ...:
----------------return 'какой-то ответ, какое-то значение'
--------if smth_to_enter_in_function <= ...:
----------------return 'какой-то другой ответ'
--------return 'сюда можно еще добавить третий вариант ответа'
2)создаем новый столбец, в котором будет указана категория
df['col_new'] = df['col_with_smth_to_enter_in_function'].apply(my_function_name)

Если записать это короче, то это будет выглядеть так:
df['col_new'] = df['col1'].apply(my_function_name)
Метод apply будет брать значение из столбца col1 для подстановки в функцию и результат, возвращаемый функцией, заносить в столбец col_new

df.apply(method, axis=1)
(чтобы применить метод к строке)