Visualization improvements

src/data_visualization.py

@@ -43,18 +43,27 @@ def main():
     cumulative_all_nbr_measurements = np.cumsum(all_nbr_measurements)
     plt.figure(figsize=(24, 9))
     plt.suptitle(visualization + ' data visualization')
-    plt.subplot(311)
+    plt.subplot(411)
     plt.plot(data_x)
     for nbr in cumulative_all_nbr_measurements:
         plt.axvline(nbr, color='r')
-    plt.subplot(312)
+    plt.ylabel('x')
+    plt.subplot(412)
     plt.plot(data_y)
     for nbr in cumulative_all_nbr_measurements:
         plt.axvline(nbr, color='r')
-    plt.subplot(313)
+    plt.ylabel('y')
+    plt.subplot(413)
     plt.plot(data_z)
     for nbr in cumulative_all_nbr_measurements:
         plt.axvline(nbr, color='r')
+    plt.ylabel('z')
+    plt.subplot(414)
+    data_mean = (np.asarray(data_x) + np.asarray(data_y) + np.asarray(data_z)) / 3.0
+    plt.plot(data_mean)
+    for nbr in cumulative_all_nbr_measurements:
+        plt.axvline(nbr, color='r')
+    plt.ylabel('mean')
     plt.show()
 
 if __name__ == "__main__":