update median calculation slightly

This commit is contained in:
Mahmoud Hashemi 2013-06-26 22:52:08 -07:00
parent 49cade467c
commit 7c0dfd35d1
1 changed files with 2 additions and 2 deletions

View File

@ -77,10 +77,10 @@ def median(vals):
if not vals: if not vals:
return 0.0 return 0.0
sorted_vals, size = sorted(vals), len(vals) sorted_vals, size = sorted(vals), len(vals)
mid = size // 2 # aka floor division
if size % 2 == 1: if size % 2 == 1:
return sorted_vals[(size - 1) / 2] return sorted_vals[mid]
else: else:
mid = size / 2
return (sorted_vals[mid - 1] + sorted_vals[mid]) / 2.0 return (sorted_vals[mid - 1] + sorted_vals[mid]) / 2.0