Python Pandas 中 DataFrame 中的天數計算？

Question

我有 DataFrame 與客戶的協議如下：

rng = pd.date_range('2020-12-01', periods=5, freq='D')
df = pd.DataFrame({ "ID" : ["1", "2", "1", "2", "2"], "Date": rng})

我需要根據上面的df計算創建新的DataFrame：

1. New1 = 從第一個協議到今天 (16.12) 的天數
2. New2 = 從上一個協議到今天 (16.12) 的天數

為了更精確，我需要創建 df 如下所示：

Answer 1

使用Series.rsub從右側減去今天，並通過 Series.dt.days 將 timedeltas 轉換為天數，然后通過Series.dt.days GroupBy.first GroupBy.agg GroupBy.last值：

now = pd.to_datetime('today')

df = (df.assign(new = df['Date'].rsub(now).dt.days)
        .groupby('ID').agg(New1 = ('new', 'first'),
                           New2 = ('new', 'last')))
        .reset_index()
print (df)
  ID  New1  New2
0  1    15    13
1  2    14    11

Answer 2

也許嘗試groupby ：

New1 = pd.to_datetime('today') - df.groupby("ID")['Date'].min()
New2 = pd.to_datetime('today') - df.groupby("ID")['Date'].max()
df2 = pd.DataFrame({'ID': df['ID'].drop_duplicates(), 'New1': New1.tolist(), 'New2': New2.tolist()})
print(df2)

Output：

  ID    New1    New2
0  1 15 days 13 days
1  2 14 days 11 days