我如何每天重置熊貓時間系列cumsum？

Question

我進口一些與時間序列數據pd.read_csv和它看起來像這樣：

     price quantity initiator 
time           
2016-07-13 16:19:31 6.20  8000   B 
2016-07-13 16:19:45 6.19  5176   S 
2016-07-13 16:25:08 6.24  15000   NaN 
2016-07-13 16:25:08 6.24  2847   S 
2016-07-13 16:25:08 6.24  39829   B 
2016-07-14 09:25:08 6.35  2398   B 
2016-07-14 09:25:08 6.30  1844   NaN 
2016-07-14 09:25:08 6.25  9538   S 
2016-07-14 09:25:08 6.15  459   B 
2016-07-14 09:25:08 6.25  1082   B 

然後我用下面的代碼來積累符合下列條件的「量」列;如果'發起人'= B/S/NaN，那麼數量是正數/負數/不計數。

instr['multipl'] = np.where(instr.initiator == 'B', 1, -1) * instr.initiator.notnull() 
instr['acc_qnty'] = (instr.quantity * instr.multipl).cumsum() 

然後我有累計數量的新列。不過，我也想每天重置/重新開始。新的一天，新的積累。

我怎麼用熊貓做到這一點？

Answer 1

IIUC你能做到這樣：

df['new'] = np.where(df.initiator == 'B', 1, -1) * df.initiator.notnull() * df.quantity 

df['result'] = df.groupby(df.index.date)['new'].cumsum() 

In [25]: df 
Out[25]: 
        price quantity initiator multipl new result 
2016-07-13 16:19:31 6.20  8000   B  1 8000 8000 
2016-07-13 16:19:45 6.19  5176   S  -1 -5176 2824 
2016-07-13 16:25:08 6.24  15000  NaN  0  0 2824 
2016-07-13 16:25:08 6.24  2847   S  -1 -2847  -23 
2016-07-13 16:25:08 6.24  39829   B  1 39829 39806 
2016-07-14 09:25:08 6.35  2398   B  1 2398 2398 
2016-07-14 09:25:08 6.30  1844  NaN  0  0 2398 
2016-07-14 09:25:08 6.25  9538   S  -1 -9538 -7140 
2016-07-14 09:25:08 6.15  459   B  1 459 -6681 
2016-07-14 09:25:08 6.25  1082   B  1 1082 -5599