将一个日期列与另一个行中的另一列进行比较

Question

我有一个dataframe df_corp ：

ID        arrival_date         leaving_date
1           01/02/20             05/02/20
2           01/03/20             07/03/20
1           12/02/20             20/02/20
1           07/03/20             10/03/20
2           10/03/20             15/03/20

我想找到一个行的 toffe_date 和到达的 到达下一个条目的与ID相对于ID 。基本上，我想知道他们再次预订多久。

因此，它看起来像这样。

ID        arrival_date         leaving_date       time_between
1           01/02/20             05/02/20             NaN
2           01/03/20             07/03/20             NaN
1           12/02/20             20/02/20              7
1           07/03/20             10/03/20             15
2           10/03/20             15/03/20              3

我已经尝试按ID进行分组来完成总和，但是我对如何从下一行获得值以及一列中的一个不同的列感到非常失落。

Answer 1

您需要转换并执行 获得上一个出发日期：

# arrival
a = pd.to_datetime(df_corp['arrival_date'], dayfirst=True)
# previous departure per ID
l = pd.to_datetime(df_corp['leaving_date'], dayfirst=True).groupby(df_corp['ID']).shift()
# difference in days
df_corp['time_between'] = (a-l).dt.days

输出：

   ID arrival_date leaving_date  time_between
0   1     01/02/20     05/02/20           NaN
1   2     01/03/20     07/03/20           NaN
2   1     12/02/20     20/02/20           7.0
3   1     07/03/20     10/03/20          16.0
4   2     10/03/20     15/03/20           3.0