Skip to content

Commit 8ad2158

Browse files
Median of Two Arrays (TheAlgorithms#3554)
* Create medianOf TwoArrays.py This code finds the median of two arrays (which may or may not be sorted initially). Example: Enter elements of an array: 1 5 4 2 Enter elements of another array: 1 7 4 2 7 The median of two arrays is : 4 * Rename medianOf TwoArrays.py to median_of _two_arrays.py * Rename median_of _two_arrays.py to median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py * Update median_of_two_arrays.py
1 parent 5fb0a70 commit 8ad2158

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

Diff for: other/median_of_two_arrays.py

+33
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
from typing import List
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)}")

0 commit comments

Comments
 (0)