def format_data(data1, data2, dim): ''' format_data is used to take data from the laser bench and return specially formatted data for use by get_slope and get_shift. This program assumes that the [:,1] dimension of the data corresponds to ring radius and that the data contains both negative and positive angles. ''' (OneInTwo, TwoInOne) = ADE.multi_where(data1[:,0], data2[:,0]) data1Overlap = data1[OneInTwo] data2Overlap = data2[TwoInOne] data1GT = data1Overlap[:,dim][np.where(data1Overlap[:,0] >= 0)] data1LT = data1Overlap[:,dim][np.where(data1Overlap[:,0] < 0)] data2GT = data2Overlap[:,dim][np.where(data2Overlap[:,0] >= 0)] data2LT = data2Overlap[:,dim][np.where(data2Overlap[:,0] < 0)] data1GT.sort() data1LT.sort() data2GT.sort() data2LT.sort() return (np.array([data1GT, data2GT]), np.array([data1LT,data2LT]))