Pandas apply function using two columns from dataframe to create a new one

Question

I have a simple function that counts weekends and holidays within two dates in python...

count_holiday_and_weekends(fromdate,todate)

How do i apply the function to create a new one in my df ?

Something like:

df['count_holiday_and_weekends'] = count_holiday_and_weekends(df['fromdate'],df['todate])

thanks in advance !

jezrael · Accepted Answer · 2017-10-25 13:33:08Z

5

Use apply with parameetr axis=1 for process by rows:

df['count_holiday_and_weekends'] = df.apply(lambda x: count_holiday_and_weekends(x['fromdate'],x['todate']), axis=1)

answered Oct 25, 2017 at 13:33

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

Comments

BENY · Accepted Answer · 2017-10-25 14:15:31Z

0

If you know which columns to be used in the function you can do

df.assign(count_holiday_and_weekends=df[['fromdate','todate']].apply(count_holiday_and_weekends,axis=1))

answered Oct 25, 2017 at 14:15

BENY

324k22 gold badges176 silver badges250 bronze badges