Calculate pandas dataframe column not using for loops

Question

I have a data frame like this,

Date        Open    High to Low   X
27-Feb-15   A           P         x1
26-Feb-15   B           Q         x2
25-Feb-15   C           R         x3
24-Feb-15   D           S         x4

i need to calculate X column values like follows,

x1 = (P+Q)/B
x2 = (Q+R)/C
   '
   '

is there any way to do this not using for loops using pandas?

jezrael · Accepted Answer · 2020-07-09 10:24:48Z

2

Use Rolling.sum with division by Series.div:

print (df)
        Date  Open  High to Low   X
0  27-Feb-15    10            1  x1
1  26-Feb-15    20            2  x2
2  25-Feb-15    50            3  x3
3  24-Feb-15   100            4  x4

df['X'] = df['High to Low'].rolling(2).sum().div(df['Open'])
print (df)
        Date  Open  High to Low     X
0  27-Feb-15    10            1   NaN
1  26-Feb-15    20            2  0.15
2  25-Feb-15    50            3  0.10
3  24-Feb-15   100            4  0.07

If necessary shift data add Series.shift:

df['X'] = df['High to Low'].rolling(2).sum().div(df['Open']).shift(-1)
print (df)
        Date  Open  High to Low     X
0  27-Feb-15    10            1  0.15
1  26-Feb-15    20            2  0.10
2  25-Feb-15    50            3  0.07
3  24-Feb-15   100            4   NaN

answered Jul 9, 2020 at 10:24

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

2 Comments

dfcsdf Over a year ago

Thank you very much jezrael ,this is exactly i wanted.

jezrael Over a year ago

@sanuja - Thanks, glad to help. Don't forget to accept the answer, if it suits you! :)

Collectives™ on Stack Overflow

Calculate pandas dataframe column not using for loops

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