Pandas使用dataframe中的两列来应用函数来创建一个新列

时间:2017-10-25 13:30:32

标签: python pandas apply

我有一个简单的函数,可以在python中的两个日期内计算周末和假期......

count_holiday_and_weekends(fromdate,todate)

如何应用该功能在我的df中创建一个新功能?

类似的东西:

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

提前感谢!

2 个答案:

答案 0 :(得分:5)

apply与parameetr axis=1一起用于按行处理:

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

答案 1 :(得分:0)

如果您知道可以在函数中使用哪些列

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