Fix UCL-RITS/rse-classwork-2020#102 and UCL-RITS/rse-classwork-2020#129

twemyss · Nov 19, 2020 · 426254c · 426254c
1 parent 02fd879
commit 426254c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sagital_brain.py b/sagital_brain.py
@@ -2,7 +2,6 @@
 
 import numpy as np
 
-
 def run_averages(file_input='brain_sample.csv', file_output='brain_average.csv'):
     """
     Calculates the average through the coronal planes
@@ -16,7 +15,7 @@ def run_averages(file_input='brain_sample.csv', file_output='brain_average.csv')
 
     # Calculates the averages through the sagital/horizontal planes
     # and makes it as a row vector
-    averages = planes.mean(axis=0)[np.newaxis, :]
+    averages = planes.mean(axis=1)[np.newaxis, :]
 
     # write it out on my file
     np.savetxt(file_output, averages, fmt='%.1f', delimiter=',')