iframe-proxy | Sunbelt Computer Software

39 lines (29 loc) · 685 Bytes

from typing import Union

def median(nums: list) -> Union[int, float]:

"""

Find median of a list of numbers.

Wiki: https://en.wikipedia.org/wiki/Median

>>> 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