Advertisement
Guest User

Untitled

a guest
Oct 22nd, 2016
54
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.39 KB | None | 0 0
  1. def smooth_moving_average(df,svar,windows_size):
  2. import numpy as np
  3.  
  4. # build windows to be moving
  5. box = np.ones(windows_size)/windows_size
  6. # calculate and store y same df
  7. df["%s_smooth%s"%(svar,windows_size)] = np.convolve(df[svar], box, mode='same')
  8. # filter the firsts lines
  9. return df[~df.index.isin(list(df.index[:windows_size]))]
  10.  
  11. WS= smooth_movingavg(WS,"ws1",3)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement