diff --git a/main.py b/main.py index 7be6ff6..6747818 100644 --- a/main.py +++ b/main.py @@ -657,17 +657,11 @@ if __name__ == '__main__': # Task 3 t_15_data = m1TruthData[0][IMU_TIME_HEADER + [TRUTH_N_POS_HEADER[1]]].iloc[(m1TruthData[0][IMU_TIME_HEADER[0]]-15).abs().argsort()[:1]].values, \ m1TruthData[2][IMU_TIME_HEADER + [TRUTH_FSP_X_HEADER[1]]].iloc[(m1TruthData[2][IMU_TIME_HEADER[0]]-15).abs().argsort()[:1]].values - - if t_15_data[0][0][0] == t_15_data[1][0][0]: - t_15_data = pd.DataFrame({ - IMU_TIME_HEADER[0] : [t_15_data[0][0][0]], - TRUTH_N_POS_HEADER[1] : [t_15_data[0][0][1]], - TRUTH_FSP_X_HEADER[1] : [t_15_data[1][0][1]] - }) - else: - print(t_15_data) - raise ValueError("Something broke :(") - + t_15_data = pd.DataFrame({ + IMU_TIME_HEADER[0] : [t_15_data[0][0][0]], + TRUTH_N_POS_HEADER[1] : [t_15_data[0][0][1]], + TRUTH_FSP_X_HEADER[1] : [t_15_data[1][0][1]] + }) print(t_15_data) # Task 4