diff --git a/main.py b/main.py index 6802368..133cd5a 100644 --- a/main.py +++ b/main.py @@ -38,13 +38,15 @@ def importIMUData(mission, imu): # If IMU is not a string, then convert based on bool eval where "H" == True if type(imu) != str: imu = "H" if imu else "L" - return pd.read_csv( + data = pd.read_csv( f"./data/IMU_M{str(mission) + str(imu)}.txt", header=None, skiprows=1, names=IMU_DATA_HEADER, dtype=[float, float, float, float, float, float, float], ) + return data + m1_IMUData = importIMUData(1, 0), importIMUData(1, 1) #(L, H) Data m2_IMUData = importIMUData(2, 0), importIMUData(2, 1)