diff --git a/t13_median1.py b/t13_median1.py index 4b5b69e..ebdfde0 100644 --- a/t13_median1.py +++ b/t13_median1.py @@ -1,12 +1,6 @@ # Find median of n numbers +from numpy import median A = [4, 1, 5, 6, 2, 5] -A.sort() print(A) - -mid = len(A) // 2 -if len(A) % 2 == 0: # even n, take average of middle 2 values - median = (A[mid-1] + A[mid]) / 2 -else: # odd n - median = A[mid] -print(median) +print(median(A))