Pandas - apply function to multiple columns

Question

There are two columns in my dataframe. One is called brand name,another is total amount

I defined a function to determine that:

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

I want to apply the function to dataframe - df[["Brand Name","Total amount"]] and it return errors.

Please advise how to solve this problem?Thanks.

Please don't use image as input example.https://stackoverflow.com/questions/20109391/how-to-make-good-reproducible-pandas-examples — Hsgao, Nov 01 '18 at 06:28
Can you show your function for determining `something` and `total amount > 'some number'`?. Also, please provide a [Minimal, Complete, and Verifiable example](https://stackoverflow.com/help/mcve). — Space Impact, Nov 01 '18 at 06:30
What errors are you getting? How did you define the function? How did you try to apply it to those two columns? — r2evans, Nov 01 '18 at 07:49

score 3 · Answer 1 · answered Nov 01 '18 at 06:31

3

You actually don't need a function. pandas provides you options to filter the dataset:

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

answered Nov 01 '18 at 06:31

Mayank Porwal

score 2 · Accepted Answer · answered Nov 01 '18 at 06:28

2

You function should look something like this:

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

Use:

df.apply(func_name, axis = 1)

answered Nov 01 '18 at 06:28

Mohit Motwani

2 Answers2