for i in range(4): for j in range(i): k+=1; plt.subplot(2,3,k); plt.plot(dt[:,j], dt[:,i], '.'); plt.title('L%d vs L%d' % (j+1,i+1)) fig.savefig(os.path.join(fig_directory, 'stage_durations_correlation.pdf')) #%% PCA on stage durations fig = plt.figure(412); plt.clf(); fplt.plot_pca(dt[:,:-1]) fig.savefig(os.path.join(fig_directory, 'stage_durations_pca.pdf')) #%% Get stage durations from Roaming Dwelling data set strain = 'N2'; rd_data = rexp.load_data(strain); rd_stage_ids = rd_data.stage_switch;
plt.hist(bend, bins=256) # resolve for stages etc mt = np.max(np.abs(theta), axis=1) idx = mt < 1 theta_red = theta[idx] theta_red.shape aplt.plot_array(theta_red.T) ### PCA plt.figure(12) plt.clf() aplt.plot_pca(theta) pca = aplt.PCA(theta) pca_comp = pca.Wt import worm.model as wm import worm.geometry as wgeo w = wm.WormModel(npoints=22) fig = plt.figure(17) plt.clf() for i in range(len(pca_comp)): tt = pca_comp[i] * 21 tta = tt - np.mean(tt) if i == 0: