如何从Pandas DataFrame中删除连续的一对相反的数字？

Question

如何删除与PANDAS DataFrame的相反符号的连续成对相等的对？

假设我有此输入数据框架，

incremental_changes = [2, -2, 2, 1, 4, 5, -5, 7, -6, 6]
df = pd.DataFrame({
    'idx': range(len(incremental_changes)),
    'incremental_changes': incremental_changes
})

   idx  incremental_changes
0    0                    2
1    1                   -2
2    2                    2
3    3                    1
4    4                    4
5    5                    5
6    6                   -5
7    7                    7
8    8                   -6
9    9                    6

我想获得以下

   idx  incremental_changes
0    0                    2
3    3                    1
4    4                    4
7    7                    7

注意：第一个2可以是IDX 0或2，这并不重要。

谢谢

Answer 1

可以分组连续数字并

import itertools

def remove_duplicates(s):
    ''' Generates booleans that indicate when a pair of ints with
        opposite signs are found.
    '''
    iter_ = iter(s)
    for (a,b) in itertools.zip_longest(iter_, iter_):
        if b is None:
            yield False
        else:
            yield a+b == 0
            yield a+b == 0

>>> mask = df.groupby(df['incremental_changes'].abs().diff().ne(0).cumsum()) \
             ['incremental_changes'] \
             .transform(remove_duplicates)

转换

>>> df[~mask]

   idx  incremental_changes
2    2                    2
3    3                    1
4    4                    4
7    7                    7

Answer 2

只需做滚动，然后我们过滤多个组合

s = df.incremental_changes.rolling(2).sum()
s = s.mask(s[s==0].groupby(s.ne(0).cumsum()).cumcount()==1)==0
df[~(s | s.shift(-1))]
Out[640]: 
   idx  incremental_changes
2    2                    2
3    3                    1
4    4                    4
7    7                    7