在行、多个键上滚动 datediff

Question

我有一个如下所示的数据框：

Part	Date
1	9/1/2021
1	9/8/2021
1	9/15/2021
2	9/1/2020
2	9/12/2020
2	9/14/2020

数据框已按部分排序，然后按日期排序。

我需要计算上一行中每个日期之间的天数。

每次遇到新的part 行时，都必须重新开始日期差异计算。

所以期望的输出是：

Part	Date	Diff
1	9/1/2021
1	9/8/2021	7
1	9/15/2021	7
2	9/1/2020
2	9/12/2020	11
2	9/14/2020	2

您将如何处理这些数据以实现所需的输出？

对此的任何帮助将不胜感激！

谢谢

Answer 1

使用groupby + diff:

df.groupby('Part').Date.diff()

0       NaT
1    7 days
2    7 days
3       NaT
4   11 days
5    2 days
Name: Date, dtype: timedelta64[ns]

如果没有Date作为时间戳，可以使用df.Date = pd.to_datetime(df.Date)进行转换。