Added stride width to the box plot.

paper.tex

@@ -820,16 +820,19 @@ \subsection*{Processed Data}
 For more insight into the difference in the unperturbed and perturbed data,
 Figure~\ref{fig:gait-cycle-stats-comparison} compares the distribution of a few
 gait cycle statistics. One can see that the perturbed strides have a much
-larger variation in frequency and length.
+larger variation in frequency and length and even larger variation in stride
+width.
 %
 \begin{figure}
   \centering
   \includegraphics{figures/unperturbed-perturbed-boxplot-comparison.pdf}
-  \cprotect\caption{Box plots of the average belt speed, stride frequency, and
-    stride length which compare unperturbed (red) and perturbed (blue) gait
-    cycles. The median is given with the box bounding the first and third
-    quartiles and the whiskers bound the range of the data. Produced by
-    \verb|src/unperturbed_perturbed_comparison.py|.}
+  % NOTE : Make sure to update these values (120, 519) if anything in the
+  % script changes.
+  \cprotect\caption{Box plots of the average belt speed, stride frequency,
+    stride length, and stride width which compare 120 unperturbed (U: red) and
+    519 perturbed (P: blue) gait cycles. The median is given with the box
+    bounding the first and third quartiles and the whiskers bound the range of
+    the data. Produced by \verb|src/unperturbed_perturbed_comparison.py|.}
   \label{fig:gait-cycle-stats-comparison}
 \end{figure}
 

src/unperturbed_perturbed_comparison.py

@@ -148,19 +148,32 @@
 # length.
 print('Generating the box plot comparison.')
 
-fig, axes = plt.subplots(1, 3)
+fig, axes = plt.subplots(1, 4)
 
 columns = ['Average Belt Speed', 'Stride Frequency', 'Stride Length']
 units = ['[m/s]', '[Hz]', '[m]']
 
-for col, ax, unit in zip(columns, axes.flatten(), units):
+for col, ax, unit in zip(columns, axes[:-1], units):
     ax.set_title('{} {}'.format(col, unit))
     u = unperturbed_gait_cycle_stats[col]
     p = perturbed_gait_data.gait_cycle_stats[col]
     sbn.boxplot([p.values, u.values], ax=ax, color=[blue, red],
                 whis='range',
-                names=['Perturbed\nN = {}'.format(num_perturbed_cycles),
-                       'Unperturbed\nN = {}'.format(num_unperturbed_cycles)])
+                names=['P', 'U'])
+
+unperturbed_stride_widths = \
+    (unperturbed_gait_cycles[:, :, 'RHEE.PosZ'] -
+     unperturbed_gait_cycles[:, :, 'LHEE.PosZ']).values.flatten()
+perturbed_stride_widths = \
+    (perturbed_gait_data.gait_cycles[:, :, 'RHEE.PosZ'] -
+     perturbed_gait_data.gait_cycles[:, :, 'LHEE.PosZ']).values.flatten()
+
+sbn.boxplot([perturbed_stride_widths, unperturbed_stride_widths],
+            ax=axes[-1],
+            color=[blue, red],
+            whis='range',
+            names=['P', 'U'])
+axes[-1].set_title('Stride Width [m]'.format(col, unit))
 
 plt.tight_layout()