iframe-proxy | Sunbelt Computer Software

35 lines (27 loc) · 581 Bytes

def median(nums):

"""

Find median of a list of numbers.

>>> median([0])

>>> median([4,1,3,2])

2.5

>>> median([2, 70, 6, 50, 20, 8, 4])

Args:

nums: List of nums

Returns:

Median.

"""

sorted_list = sorted(nums)

length = len(sorted_list)

mid_index = length >> 1

return (

(sorted_list[mid_index] + sorted_list[mid_index - 1]) / 2

if length % 2 == 0

else sorted_list[mid_index]

)

def main():

import doctest

doctest.testmod()

if __name__ == "__main__":

main()

Sunbelt Computer Software