熊猫基于日期添加列分组

Question

我正在尝试添加一列，以在接下来的1,2,3,4周内标记带有结束日期的那些订单项，但是使用以下代码只是覆盖了自身（这很有意义）。有没有简单的方法可以添加此分组？

本质上，用Excel来讲，我试图做一个嵌入式IF语句（IF Date

sup['Availability'] = sup['End Date'].apply(lambda x: ' Now Available' if x < date.today() else sup['End Date'])
sup['Availability'] = sup['End Date'].apply(lambda x: '1 Week' if x < date.today() + timedelta(weeks=1) else sup['First Available Date'])
sup['Availability'] = sup['End Date'].apply(lambda x: '2 Week' if x < date.today() + timedelta(weeks=2) else sup['First Available Date'])
sup['Availability'] = sup['End Date'].apply(lambda x: '3 Week' if x < date.today() + timedelta(weeks=3) else '4 Weeks')