From 7e38a29bd77bbc9d962f755c51afd69fb3878551 Mon Sep 17 00:00:00 2001 From: Cal Wing <20716204+calw20@users.noreply.github.com> Date: Thu, 2 Nov 2023 20:52:32 +1000 Subject: [PATCH] Quick refecator --- main.py | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) 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