在相同的pandas数据框中减去不同时间的值

时间:2015-05-04 18:23:27

标签: python pandas

我有一个像这样的时间序列:

            value     identifier
2007-01-01  0.781611      55
2007-01-01  0.766152      56
2007-01-01  0.766152      57
2007-01-03  0.733152      57
2007-01-12  0.766152      57
2007-02-01  0.705615      55
2007-02-01  0.032134      56 
2007-02-07  0.132134      56 
2007-02-01  0.032134      57
2008-01-01  0.026512      55
2008-01-01  0.993124      56
2008-01-01  0.993124      57
2008-01-11  0.226512      56
2008-02-01  0.226420      55
2008-02-01  0.033860      56
2008-02-01  0.033860      57
2008-02-14  0.333860      55
2008-02-24  0.334860      57

现在我想用不同的时间增量来计算差异。我的第一个方法是

last_date = '2008-02-30'
dt = datetime.timedelta(days=30)
delta_30d = datetime.datetime.strptime(last_date, "%Y-%m-%d") - dt
df.loc[last_date]['value'] - df.loc[delta_30d]['value']

但如果时间增量上的完全相同的日期不存在,我会

'the label [2008-01-30 00:00:00] is not in the [index]'

所以这是不行的。如何从请求的时间增量中获取最接近的日期并计算差异?

1 个答案:

答案 0 :(得分:1)

如果您的索引是唯一的,则可以使用DateTimeIndex.asof,如question所示。

另一种方式(来自同一个问题)将使用numpy.argmin来查找最接近的日期。例如,你可以做

In [138]: import numpy as np

In [139]: last_date = pd.Timestamp('2008-01-30')

In [140]: delta_30d = last_date - pd.Timedelta(30, 'D')

In [142]: np.argmin(np.abs(df.index - delta_30d))
Out[142]: 13

In [144]: df.iloc[13, :]
Out[144]: 
value          0.22642
identifier    55.00000
Name: 2008-02-01 00:00:00, dtype: float64

In [145]: df.iloc[np.argmin(np.abs(df.index - last_date))]['value'] - df.iloc[np.argmin(np.abs(df.index - delta_30))]['value']
Out[145]: 0.19990799999999997