diff --git a/stonesoup/plotter.py b/stonesoup/plotter.py index a7ee8741f..841e1c462 100644 --- a/stonesoup/plotter.py +++ b/stonesoup/plotter.py @@ -310,9 +310,10 @@ def plot_tracks(self, tracks, mapping, uncertainty=False, particle=False, track_ else: not_update_indexes.append(n) - data = np.array( + data = np.concatenate( [(getattr(state, 'mean', state.state_vector)[mapping, :]) - for state in track]).squeeze().T + for state in track], + axis=1) line = self.ax.plot( *data,