Create running_median_integers.py

Author: Amogh Singhal

running_median_integers.py

@@ -0,0 +1,30 @@
+# Problem: Find Median from Data Stream
+
+def findMedian(stream):
+  # print(stream)
+  streamSize = len(stream)
+
+  if streamSize == 1:
+    return stream[0]
+  else:
+    stream = sorted(stream)
+    midPt = streamSize // 2
+
+    if streamSize % 2 == 1:
+      return stream[midPt]
+    else:
+      return (stream[midPt] + stream[midPt-1]) // 2
+
+def findRunningMedian(inputArray):
+  medianArray = []
+
+  for i in range(0,len(inputArray)):
+    # print(inputArray)
+    currentMedian = findMedian(inputArray[0:i+1])
+    medianArray.append(currentMedian)
+  
+  return medianArray
+
+
+solution = findRunningMedian([1, 2, 3, 4, 5])
+print(solution)