Hackerrank 10 Days of statistics, Day 0: Mean, Median, and Mode

mean_median_mode.py

from scipy.stats import mode

raw_input();
ARR = raw_input();

def median(nums):
    nums = sorted(nums)
    
    if len(nums) % 2 == 0:
        # if even, grab the two mid letters
        return (nums[int(len(nums) / 2) - 1] + nums[int(len(nums) / 2)]) / 2.0
    else:
        return nums[int(len(nums) / 2)]

def mean(nums):
    return sum(nums) / float(len(nums))

ARR = [int(i) for i in ARR.split()] #splits the Array and make each value an interger

print mean(ARR)
print median(ARR)
print mode(ARR)[0][0]