cal/AERO4200-Ass3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Quick refecator

Browse Source
This commit is contained in:
Cal Wing 2023-11-02 20:52:32 +10:00
parent cbb0cf0452
commit 7e38a29bd7
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
main.py
View File

@ -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