Tweak mpc comparison script

scripts/crocoddyl_eval/test_6/compare_mpcs.py

@@ -58,12 +58,14 @@ def compute_RMSE(array, norm):
 ##############
 
 # [Linear, Non Linear, Planner, OSQP]
-MPCs = [True, True, True, True] # Boolean to choose which MPC to plot
+MPCs = [True, True, True, False] # Boolean to choose which MPC to plot
 MPCs_names = ["No FF", "-afeet", "JT fmpc + M ddq", "Free flyer + Tasks"]
 name_files = ["data_2021_08_27_16_08_0.npz", "data_2021_08_27_15_53_0.npz", "data_2021_08_27_15_42_0.npz", "data_2021_08_27_15_44_0.npz"] # Names of the files
 name_files = ["data_2021_09_02_16_51_0.npz", "data_2021_09_02_16_53_0.npz", "data_2021_09_02_16_55_0.npz", "data_2021_09_02_17_52_0.npz"] # Names of the files
 folder_path = "" # Folder containing the 4 .npz files
 
+name_files = ["data_2021_09_16_18_08_0.npz", "data_2021_09_16_18_12_0.npz", "data_2021_09_16_18_20_0.npz", "data_2021_09_16_18_12_0.npz"]
+
 # Common data shared by 4 MPCs
 params = lqrw.Params()  # Object that holds all controller parameters
 logs = np.load(folder_path + name_files[0])