Сравнение строк и предыдущей строки при изменении в пандами dataframe

Question

У меня продольные данные следующего вида

import pandas as pd 

df = pd.DataFrame({ 
    'a': ['apples', 'plums', 'pears', 'pears', 'pears'], 
    'b': ['grapes', 'grapes', 'grapes', 'grapes', 'bananas'], 
    'c': [0, 0, 1, 0, 1] 
}) 

и функцию, которая сравнивает списки (подробности этого не важны)

def compare(old_fruit, new_fruit): 
    if set(new_fruit) - set(old_fruit) == {'pears'}: 
     return 1 
    else: 
     return 0 

c - 1, когда меняются изменения в a и b, которые меня интересуют. Я хочу найти строки, где c = 1, захватить значения a и b в этот момент, плюс t he значения a и b из предыдущей строки, сравнить их с помощью моей функции и добавить новую серию в dataframe, показывающую результат сравнения.

Для приведенного выше примера, моя требуемая операция выполнения будет compare(['plums', 'grapes'], ['pears', 'grapes']) и compare(['pears', 'grapes'], ['pears', 'bananas']) и добавьте серии [0, 0, 1, 0, 0] к dataframe, т.е. требуемый выход является dataframe следующим образом:

pd.DataFrame({ 
    'a': ['apples', 'plums', 'pears', 'pears', 'pears'], 
    'b': ['grapes', 'grapes', 'grapes', 'grapes', 'bananas'], 
    'c': [0, 0, 1, 0, 1], 
    'd': [0, 0, 1, 0, 0] 
}) 

Answer 1

Делайте то, что вы хотите сравнить в векторном виде:

df_set = df[['a', 'b']].apply(set, axis=1) 

df_set 
Out[38]: 
0 {grapes, apples} 
1  {grapes, plums} 
2  {grapes, pears} 
3  {grapes, pears} 
4 {bananas, pears} 
dtype: object 

(df_set - df_set.shift()) == {'pears'} 
Out[39]: 
0 False 
1 False 
2  True 
3 False 
4 False 
dtype: bool