Skip to content

Commit a7b4fc6

Browse files
Update mean_var_std.py
1 parent 43f8af7 commit a7b4fc6

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed
Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,15 @@
11
import numpy as np
22

3-
def calculate(list):
4-
calculations = None
5-
6-
7-
8-
9-
return calculations
3+
def calculate(arr):
4+
if len(arr) != 9:
5+
raise ValueError("List must contain nine numbers.")
6+
matrix1 = np.array(arr)
7+
matrix2 = matrix1.reshape(3, 3)
8+
calculations = {"mean": [list(np.mean(matrix2, axis=0)), list(np.mean(matrix2, axis=1)), np.mean(matrix1)],
9+
"variance": [list(np.var(matrix2, axis=0)), list(np.var(matrix2, axis=1)), np.var(matrix1)],
10+
"standard deviation": [list(np.std(matrix2, axis=0)), list(np.std(matrix2, axis=1)), np.std(matrix1)],
11+
"max": [list(np.max(matrix2, axis=0)), list(np.max(matrix2, axis=1)), np.max(matrix1)],
12+
"min": [list(np.min(matrix2, axis=0)), list(np.min(matrix2, axis=1)), np.min(matrix1)],
13+
"sum": [list(np.sum(matrix2, axis=0)), list(np.sum(matrix2, axis=1)), np.sum(matrix1)]
14+
}
15+
return calculations

0 commit comments

Comments
 (0)