def decleration_color_coded_for_post(dataset, plot=True): fa.calc_stats_at_deceleration(dataset, keys=None, time_offset=0, return_vals=False) angleok = np.where( (dataset.angle_at_deceleration < 2)*(dataset.angle_at_deceleration > .01) )[0].tolist() fig = plt.figure() ax = fig.add_subplot(111) black_post = [] checkered_post = [] for i, p in enumerate(dataset.post_type_at_deceleration): if i in angleok: if 'black' in p: black_post.append(i) if 'checkered' in p: checkered_post.append(i) print '*'*40 print len(black_post), len(checkered_post) bins, hists, hist_std, curves = floris.histogram(ax, [np.log(dataset.angle_at_deceleration[black_post]), np.log(dataset.angle_at_deceleration[checkered_post])], bins=16, colors=['black', 'teal'], bin_width_ratio=0.9, edgecolor='none', bar_alpha=0.2, curve_fill_alpha=0, curve_line_alpha=0.8, return_vals=True, normed_occurences='total', bootstrap_std=True) ax.plot(np.log(dataset.angle_at_deceleration[black_post]), dataset.speed_at_deceleration[black_post], '.', color='black', alpha=1) ax.plot(np.log(dataset.angle_at_deceleration[checkered_post]), dataset.speed_at_deceleration[checkered_post], '.', color='teal', alpha=1) #ax.set_ylim(0,1.2) fa.fix_angle_log_spine(ax, histograms=True) fig.savefig('deceleration_color_coded.pdf', format='pdf')
def t_test_speed_at_deceleration(dataset): fa.calc_stats_at_deceleration(dataset, keys=None, time_offset=0, return_vals=False) angleok = np.where( (dataset.angle_at_deceleration < 2)*(dataset.angle_at_deceleration > .01) )[0].tolist() black_post = [] checkered_post = [] for i, p in enumerate(dataset.post_type_at_deceleration): if i in angleok: if 'black' in p: black_post.append(i) if 'checkered' in p: checkered_post.append(i) print '*'*40 print len(black_post), len(checkered_post) print scipy.stats.ttest_ind(dataset.speed_at_deceleration[black_post], dataset.speed_at_deceleration[checkered_post]) print 'black: ', np.mean(dataset.speed_at_deceleration[black_post]), 'n: ', len(dataset.speed_at_deceleration[black_post]) print 'checkered: ', np.mean(dataset.speed_at_deceleration[checkered_post]), 'n: ', len(dataset.speed_at_deceleration[checkered_post]) print 'total n - 2: ', len(black_post) + len(checkered_post) - 2