Median of Two Arrays

other/median_of_two_arrays.py

@@ -0,0 +1,33 @@
+from typing import List
+
+
+def median_of_two_arrays(nums1: List[float], nums2: List[float]) -> float:
+    """
+    >>> median_of_two_arrays([1, 2], [3])
+    2
+    >>> 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])
+    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)}")