Pandas: appending column with condition

Question

I have a sample time series data (stock) as below:

                 Date  PX_OPEN  PX_LAST
Date                                   
2011-01-03 2011-01-03    31.18    31.26
2011-01-04 2011-01-04    31.42    31.02
2011-01-05 2011-01-05    31.10    30.54
2011-01-06 2011-01-06    30.66    30.54
2011-01-07 2011-01-07    31.50    30.66
2011-01-10 2011-01-10    30.82    30.94

I would like to add a new column GAP based on the following conditions:

If current day open is higher than previous day last, then GAP = up.
If current day open is lower than previous day last, then GAP = down.
Otherwise, GAP = unch. (Alternatively, up can be changed to +1, down to -1, and unch to 0.)

I can do this with if and for loop, but that would defeat the efficiency of verctorized operation in Pandas. Can anyone help?

IanS · Accepted Answer · 2016-12-12 15:12:51Z

2

Use nested np.where calls:

import numpy as np
df['GAP'] = np.where(df['PX_OPEN'] > df['PX_LAST'].shift(), 'up',
            np.where(df['PX_OPEN'] < df['PX_LAST'].shift(), 'down', 'unch'))

answered Dec 12, 2016 at 15:12

IanS

16.3k9 gold badges64 silver badges87 bronze badges

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

Collectives™ on Stack Overflow

Pandas: appending column with condition

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