python - 对数据框上的列中的日期时间进行计算

Question

假设我在数据框中有两列，由事件的概率和剩余时间组成。

      prob          time         
0   0.975909   0 days 00:00:00   
1   0.957819   0 days 01:00:00   
2   0.937498   0 days 02:00:00   
3   0.912779   0 days 03:00:00   
4   0.894139   0 days 04:00:00   
5   0.873184   0 days 05:00:00   
6   0.847748   0 days 06:00:00   
7   0.828572   0 days 07:00:00   
8   0.807029   0 days 08:00:00   
9   0.780847   0 days 09:00:00   
10  0.761082   0 days 10:00:00   
11  0.738855   0 days 11:00:00   
12  0.711733   0 days 12:00:00   

我想计算准确的时间和日期，一些额外的输入是日期和时间，以及概率，例如我会这样:

# Type the date of input data 
i = datetime.datetime.now() #e.g. 2018-01-01 00:00:00

# Type the expected probability 
exprob = 0.80

我需要的输出结果是:找到与 'exprob' (0.80) --> 0.80709 最接近的概率，然后计算 'i' + 与 0.80709 相关的时间 = 2018-01-01 08:00:00

Answer 1

您可以使用idxmin查找df['prob']和exprob之间的最小差异的索引，然后找到Timedelta 并将其添加到日期 i 中，如下所示:

i = datetime.datetime.now()
exprob = 0.80

df.loc[((df['prob'] - exprob).abs().idxmin()),'time'] + i
Timestamp('2018-11-08 18:36:11.529609')