Advertisement
Guest User

Untitled

a guest
May 23rd, 2019
70
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.37 KB | None | 0 0
  1. # Windowed time lagged cross correlation
  2. seconds = 5
  3. fps = 30
  4. no_splits = 20
  5. samples_per_split = df.shape[0]/no_splits
  6. rss=[]
  7. for t in range(0, no_splits):
  8. d1 = df['S1_Joy'].loc[(t)*samples_per_split:(t+1)*samples_per_split]
  9. d2 = df['S2_Joy'].loc[(t)*samples_per_split:(t+1)*samples_per_split]
  10. rs = [crosscorr(d1,d2, lag) for lag in range(-int(seconds*fps-1),int(seconds*fps))]
  11. rss.append(rs)
  12. rss = pd.DataFrame(rss)
  13. f,ax = plt.subplots(figsize=(10,5))
  14. sns.heatmap(rss,cmap='RdBu_r',ax=ax)
  15. ax.set(title=f'Windowed Time Lagged Cross Correlation',xlim=[0,300], xlabel='Offset',ylabel='Window epochs')
  16. ax.set_xticklabels([int(item-150) for item in ax.get_xticks()]);
  17.  
  18. # Rolling window time lagged cross correlation
  19. seconds = 5
  20. fps = 30
  21. window_size = 300 #samples
  22. t_start = 0
  23. t_end = t_start + window_size
  24. step_size = 30
  25. rss=[]
  26. while t_end < 5400:
  27. d1 = df['S1_Joy'].iloc[t_start:t_end]
  28. d2 = df['S2_Joy'].iloc[t_start:t_end]
  29. rs = [crosscorr(d1,d2, lag, wrap=False) for lag in range(-int(seconds*fps-1),int(seconds*fps))]
  30. rss.append(rs)
  31. t_start = t_start + step_size
  32. t_end = t_end + step_size
  33. rss = pd.DataFrame(rss)
  34.  
  35. f,ax = plt.subplots(figsize=(10,10))
  36. sns.heatmap(rss,cmap='RdBu_r',ax=ax)
  37. ax.set(title=f'Rolling Windowed Time Lagged Cross Correlation',xlim=[0,300], xlabel='Offset',ylabel='Epochs')
  38. ax.set_xticklabels([int(item-150) for item in ax.get_xticks()]);
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement