Добрый день! Пишу функцию для создания категорий
Принцип работы описываю ниже, но мой код не работает
Есть два столбца с типом данных int64.
На основании оценки столбца ‘col2’ и сравнения его значений с ‘col1’ нужно заполнить значения для нового столбца ‘category’.
Скрин
что-то вроде
def cat_func(smth):
if col2 == 1:
return 'один’
if col2 == df[df]:
return ‘равен’
return ‘не равен’
df = df.apply(cat_func)
Пробую и с where реализовать, и с loc, и тип данных пробую менять предварительно, но не получается
Как лучше сделать?