TheAlgorithms-Python

1
from __future__ import annotations
2

3

4
def median_of_two_arrays(nums1: list[float], nums2: list[float]) -> float:
5
    """
6
    >>> median_of_two_arrays([1, 2], [3])
7
    2
8
    >>> median_of_two_arrays([0, -1.1], [2.5, 1])
9
    0.5
10
    >>> median_of_two_arrays([], [2.5, 1])
11
    1.75
12
    >>> median_of_two_arrays([], [0])
13
    0
14
    >>> median_of_two_arrays([], [])
15
    Traceback (most recent call last):
16
      ...
17
    IndexError: list index out of range
18
    """
19
    all_numbers = sorted(nums1 + nums2)
20
    div, mod = divmod(len(all_numbers), 2)
21
    if mod == 1:
22
        return all_numbers[div]
23
    else:
24
        return (all_numbers[div] + all_numbers[div - 1]) / 2
25

26

27
if __name__ == "__main__":
28
    import doctest
29

30
    doctest.testmod()
31
    array_1 = [float(x) for x in input("Enter the elements of first array: ").split()]
32
    array_2 = [float(x) for x in input("Enter the elements of second array: ").split()]
33
    print(f"The median of two arrays is: {median_of_two_arrays(array_1, array_2)}")
34