Use shift function in pandas Dataframe function

Question

I want to create a function which compares different rows within a Pandas dataframe.
My current function looks like this:

def f(row):
    if row['A'].shift(1) == row['B']:
        val = 0        
    else:
        val = 1
    return val

I receive the following error message:

AttributeError: ("'numpy.float64' object has no attribute 'shift'", 'occurred at index 2006-02-28 00:00:00')

I understand that the problem occurs in the first row of the dataframe because there is no row which can be shifted. Does anyone know how to rewrite the funtion? Any help is appreciated.

BENY · Accepted Answer · 2018-09-05 13:35:59Z

1

With pandas you can .

(~(df.A.shift()==df.B)).astype(int) # since default of shift is 1

answered Sep 5, 2018 at 13:35

BENY

324k22 gold badges176 silver badges250 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Use shift function in pandas Dataframe function

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related