python - Pandas 添加基于日期的列分组

Question

我正在尝试添加一个列，在接下来的 1、2、3、4 周内用结束日期标记这些行项目，但使用下面的代码只是覆盖了自身（这是有道理的）。有没有简单的方法来添加这个分组？

本质上，在 Excel 术语中，我正在尝试执行嵌入式 IF 语句（IF Date < TODAY() , X , IF Date < TODAY() + 7, Y, IF ...）

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')

标签： pythonpandas