groupby с интервалами с перекрывающимися интервалами

Question

У меня есть временной ряд в объекте dataframe python pandas, и я хочу создать группу на основе индекса, но я хочу, чтобы перекрывающиеся группы группы были не различны. Header_sec является столбцом индекса. Каждая группа состоит из 2-секундного окна. Input dataFrame

header_sec 
1 17004 days 22:17:13 
2 17004 days 22:17:13 
3 17004 days 22:17:13 
4 17004 days 22:17:13 
5 17004 days 22:17:14 
6 17004 days 22:17:14 
7 17004 days 22:17:14 
8 17004 days 22:17:14 
9 17004 days 22:17:15 
10 17004 days 22:17:15 
11 17004 days 22:17:15 
12 17004 days 22:17:15 
13 17004 days 22:17:16 
14 17004 days 22:17:16 
15 17004 days 22:17:16 
16 17004 days 22:17:16 
17 17004 days 22:17:17 
18 17004 days 22:17:17 
19 17004 days 22:17:17 
20 17004 days 22:17:17 

Моя первая группа должна иметь

1 17004 days 22:17:13 
2 17004 days 22:17:13 
3 17004 days 22:17:13 
4 17004 days 22:17:13 
5 17004 days 22:17:14 
6 17004 days 22:17:14 
7 17004 days 22:17:14 
8 17004 days 22:17:14 

Вторая группа начинается с предыдущего индекса и занимает 1/2 записей в предыдущем секунду.

7 17004 days 22:17:14 
8 17004 days 22:17:14 
9 17004 days 22:17:15 
10 17004 days 22:17:15 
11 17004 days 22:17:15 
12 17004 days 22:17:15 
13 17004 days 22:17:16 
14 17004 days 22:17:16 

Третья группа .....

13 17004 days 22:17:16 
14 17004 days 22:17:16 
15 17004 days 22:17:16 
16 17004 days 22:17:16 
17 17004 days 22:17:17 
18 17004 days 22:17:17 
19 17004 days 22:17:17 
20 17004 days 22:17:17 

Если я GroupBy по индексу,

dfgroup=df.groupby(df.index) 

это дает одну группу в секунду. Что было бы лучшим способом объединить эти группы?

Answer 1

Вот техник:

import numpy as np # if you have not already done this 

grouped = df.groupby(df.index) 

for name, group in grouped: 
    try: 
     prev_sec = df.loc[(name - pd.to_timedelta(1, unit='s')), :] 
    except KeyError: 
     prev_sec = pd.DataFrame(columns=group.columns) 
    try: 
     next_sec = df.loc[(name + pd.to_timedelta(1, unit='s')), :] 
    except KeyError: 
     next_sec = pd.DataFrame(columns=group.columns) 
    Pn = 2 # replace this with int(len(prev_sec)/2) to get half rows from previous second 
    Nn = 2 # replace this with int(len(next_sec)/2) to get half rows from next second 
    group = pd.concat([prev_sec.iloc[-Pn:,:], group, next_sec.iloc[:Nn,:]]) 

    # Replace the below lines with your operations 
    print(name, group)