Plot path histogram in main without individual trajectories

Author: plainerman

main.py

@@ -236,6 +236,9 @@ def main():
         plot_energy(system, [x_t_stoch_no_vel[0], x_t_stoch_no_vel[-1]], args.log_plots)
         show_or_save_fig(args.save_dir, 'path_energy_stochastic', args.extension)
 
+        system.plot(title='Stochastic Paths', trajectories=x_t_stoch_no_vel)
+        show_or_save_fig(args.save_dir, 'paths_stochastic', args.extension)
+
         system.plot(title='Stochastic Paths', trajectories=x_t_stoch_no_vel)
         trajectories_to_plot = 2
         colors = plt.rcParams['axes.prop_cycle'].by_key()['color']