Shift rows in a Column when equal to specific value

Question

I want to shift rows in a pandas df when values are equal to a specific value in a Column. For the df below, I'm trying to shift the values in Column B to Column A when values in A == x.

import pandas as pd

df = pd.DataFrame({
    'A' : [1,'x','x','x',5],
    'B' : ['x',2,3,4,'x'],
        })

This is my attempt:

df = df.loc[df.A.shift(-1) == df.A.shift(1), 'x'] = df.A.shift(1)

Intended Output:

anky · Accepted Answer · 2019-03-07 06:09:39Z

4

You can use:

m = df.A.eq('x')
df[m]=df[m].shift(-1,axis=1)
print(df)

   A    B
0  1    x
1  2  NaN
2  3  NaN
3  4  NaN
4  5    x

edited Mar 7, 2019 at 6:09

answered Mar 7, 2019 at 5:52

anky

75.3k11 gold badges46 silver badges76 bronze badges

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

Comments

heena bawa · Accepted Answer · 2019-03-07 05:59:46Z

2

You can use:

df[df.A=='x'] = df.shift(-1,axis=1)

print(df)

   A    B
0  1    x
1  2  NaN
2  3  NaN
3  4  NaN
4  5    x

answered Mar 7, 2019 at 5:59

heena bawa

8286 silver badges5 bronze badges

Collectives™ on Stack Overflow

Shift rows in a Column when equal to specific value

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related