Панды - применить функцию к нескольким столбцам

Question

В моем фрейме данных есть два столбца. Один называется brand name, другой - total amount

Я определил функцию, чтобы определить, что:

if the brand name == 'something' and total amount > 'some number'
then return 'something'

Я хочу применить функцию к фрейму данных - df[["Brand Name","Total amount"]], и она возвращает ошибки.

Посоветуйте, пожалуйста, как решить эту проблему? Спасибо.

Answer 1

Вам на самом деле не нужна функция. pandas предоставляет вам варианты filter набора данных:

df[(df[brand name] == 'something') & (df[total amount] > 'some number')]

Answer 2

Ваша функция должна выглядеть примерно так:

def func_name(x):
    if x['Brand Name'] == 'something' and x['total amount'] > y:
        return 'some'

Использование:

df.apply(func_name, axis = 1)