Pandas的apply函数如何使用数据框中的两列创建一个新列？

Question

我有一个简单的功能，用于在Python中计算两个日期之间的周末和假期的天数。您想如何将此函数应用于创建新的数据框呢？可以使用以下代码：

df ['count_holiday_and_weekends'] = count_holiday_and_weekends（df ['fromdate']，df ['todate']）

请注意，这将在您的数据框中添加一个名为“count_holiday_and_weekends”的列，并将该列填充为从“fromdate”到“todate”的周末和假期天数。

感谢您的提问！

- user6771469

Answer 1

使用参数axis=1与apply一起对行进行处理：

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

- jezrael

Answer 2

如果您知道在函数中要使用哪些列，您可以执行以下操作

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

- BENY