def plot_naksu_by_sid_and_bend(): data = get_merged_data() ci = 3 pdf_out = PdfPages('/tmp/naksu_by_sid_and_bend_'+str(ci)+'.pdf') for sid in SESSIONS: sdata = data[ data['session_id'] == sid] plt.figure() plt.title(sid) for lap in np.unique(sdata['lap']): ldata = sdata[ sdata['lap'] == lap ] segdata = get_segment_data(sid, CORNERING[ci], ldata) plt.plot(segdata['dist'], segdata['naksu_x'], '-g') pdf_out.savefig() plt.close() pdf_out.close()
def head_movement(): data = get_merged_data() stats = [] for sid in SESSIONS: sd = data[data['session_id'] == sid] gox = sd['g_origin_x'] goy = sd['g_origin_y'] m = (np.mean(gox), np.mean(goy)) s = (np.std(gox), np.std(goy)) stats.append((m,s)) print sid print m[0], m[1] print s[0], s[1] stats = np.array(stats) m = stats[:,0] s = stats[:,1] print 'total:' print np.mean(m[:,0]), np.mean(m[:,1]) print np.mean(s[:,0]), np.mean(s[:,1]) """