Python Pandas : groupby의 이전 행에있는 값이 특정 조건을 충족하는 경우 데이터 프레임에서 행 제거

Question

일주일에 1을 초과하여 비 순차적이되면 groupby에서 데이터를 제거하려고합니다. 예 : If 일주일에 간격이 있으면 해당 행과 그 행의 후속 행을 제거하고 싶습니다. 아래는 내가 가지고있는 데이터의 구조와 내가 찾고있는 이상적인 결과의 간단한 예제이다. 데이터는 국가 및 제품별로 그룹화해야합니다. 주 5 위의 주 또한 DE 애플을 제거이 3 주이기 때문에 앞의 주 주 5 & 6 제거 DE 애플 4. 때문에

import pandas as pd 

data = {'Country' : ['US','US','US','US','US','DE','DE','DE','DE','DE'],'Product' :   ['Coke','Coke','Coke','Coke','Coke','Apple','Apple','Apple','Apple','Apple'],'Week' : [1,2,3,4,6,1,2,3,5,6] } 

df = pd.DataFrame(data) 

print df 

#Current starting Dataframe. 
    Country Product Week 
0  US Coke  1 
1  US Coke  2 
2  US Coke  3 
3  US Coke  4 
4  US Coke  6 
5  DE Apple  1 
6  DE Apple  2 
7  DE Apple  3 
8  DE Apple  5 
9  DE Apple  6 

#Ideal Output below: 
    Country Product Week 
0  US Coke  1 
1  US Coke  2 
2  US Coke  3 
3  US Coke  4 
5  DE Apple  1 
6  DE Apple  2 
7  DE Apple  3 

그래서 출력은 미국 콜라를 위해 6 주를 제거 그 5 또는 1

Answer 1

의 DIFF()되어 이전이

df.groupby(['Country', 'Product']).apply(lambda sdf: sdf[(sdf.Week.diff(1).fillna(1) != 1).astype('int').cumsum() == 0]).reset_index(drop=True) 

또 다른 방법을 작동해야에도 불구하고 6 주, 즉 즉 주 연속 세트를 생성하고 체크 대상 (더 읽을 수 있습니다 관측하다 에드 주)

df['expected_week'] = df.groupby(['Country', 'Product']).Week.transform(lambda s: range(s.min(), s.min() + s.size)) 
df[df.Week == df.expected_week] 

Answer 2

당신은이 방법을 시도해 볼 수도 있습니다 ...

def eliminate(x): 
    x['g'] = x['Week'] - np.arange(x.shape[0]) 
    x = x[x['g'] == x['g'].min()] 
    x = x.drop('g',1) 

    return x 

out = df.groupby('Product').apply(eliminate).reset_index(level=0,drop=True)