iframe-proxy | Sunbelt Computer Software

33 lines (28 loc) · 989 Bytes

from __future__ import annotations

def median_of_two_arrays(nums1: list[float], nums2: list[float]) -> float:

"""

>>> median_of_two_arrays([1, 2], [3])

>>> median_of_two_arrays([0, -1.1], [2.5, 1])

0.5

>>> median_of_two_arrays([], [2.5, 1])

1.75

>>> median_of_two_arrays([], [0])

>>> median_of_two_arrays([], [])

Traceback (most recent call last):

...

IndexError: list index out of range

"""

all_numbers = sorted(nums1 + nums2)

div, mod = divmod(len(all_numbers), 2)

if mod == 1:

return all_numbers[div]

else:

return (all_numbers[div] + all_numbers[div - 1]) / 2

if __name__ == "__main__":

import doctest

doctest.testmod()

array_1 = [float(x) for x in input("Enter the elements of first array: ").split()]

array_2 = [float(x) for x in input("Enter the elements of second array: ").split()]

print(f"The median of two arrays is: {median_of_two_arrays(array_1, array_2)}")

Sunbelt Computer Software