How do you shift each row in pandas data frame by a specific value?

Question

If I have a pandas dataframe like this:

 2  3   4  NaN NaN  NaN
 1 NaN NaN NaN NaN  NaN
 5  6   7   2   3   NaN
 4  3  NaN NaN NaN  NaN

and an array for the number I would like to shift: array = [2, 4, 0, 3]

How do I iterate through each row to shift the columns by the number in my array to get something like this:

 NaN  NaN   2   3   4   NaN
 NaN  NaN  NaN  NaN 1   NaN
  5    6    7   2   3   NaN
 NaN  NaN  NaN  3   4   NaN

I was trying to do something like this but had no luck.

 df = pd.DataFrame(values)
 for rows in df.iterrows():
   df[rows] = df.shift[change_in_bins[rows]]

Is the last column always NaN?

cs95
– cs95

2018-11-12 04:48:16 +00:00
Commented Nov 12, 2018 at 4:48 — cs95
– cs95, Commented Nov 12, 2018 at 4:48

Space Impact · Accepted Answer · 2018-11-12 04:52:25Z

3

Use for loop with loc and shift:

for index,value in enumerate([2, 4, 0, 3]):
    df.loc[index,:] = df.loc[index,:].shift(value)

print(df)
     0    1    2    3    4   5
0  NaN  NaN  2.0  3.0  4.0 NaN
1  NaN  NaN  NaN  NaN  1.0 NaN
2  5.0  6.0  7.0  2.0  3.0 NaN
3  NaN  NaN  NaN  4.0  3.0 NaN

edited Nov 12, 2018 at 4:52

answered Nov 12, 2018 at 4:44

Space Impact

13.3k26 silver badges51 bronze badges

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

2 Comments

Zmann3000 Over a year ago

Thank you so much!

Space Impact Over a year ago

@TroyZaremba You are Welcome!

Collectives™ on Stack Overflow

How do you shift each row in pandas data frame by a specific value?

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related