Python-сообщество

Добрый день! Пишу функцию для создания категорий
Принцип работы описываю ниже, но мой код не работает

Есть два столбца с типом данных 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, и тип данных пробую менять предварительно, но не получается
Как лучше сделать?

Отредактировано alexo_k (Дек. 5, 2020 00:12:49)

Скрин

Прикреплённый файлы:
2020-12-05 01-09-02 Без имени 5 - LibreOffice Calc.png (1,7 KБ)

alexo_k
Принцип работы описываю ниже, но мой код не работает
Есть два столбца с типом данных int64.