如何使用 Python 堆实现动态中位数问题？

import heapq

class DynamicMedian:
    def __init__(self):
        self.max_heap = []  # 最大堆，存储数列中较小的一半数字
        self.min_heap = []  # 最小堆，存储数列中较大的一半数字

    def insert(self, num):
        if not self.max_heap or num <= -self.max_heap[0]:
            heapq.heappush(self.max_heap, -num)
        else:
            heapq.heappush(self.min_heap, num)

        # 平衡两个堆的大小，使得差值不大于 1
        if len(self.max_heap) - len(self.min_heap) > 1:
            temp = -heapq.heappop(self.max_heap)
            heapq.heappush(self.min_heap, temp)
        elif len(self.min_heap) - len(self.max_heap) > 1:
            temp = heapq.heappop(self.min_heap)
            heapq.heappush(self.max_heap, -temp)

    def find_median(self):
        # 两个堆中的元素个数相同时，中位数为两个堆顶元素的平均值
        if len(self.max_heap) == len(self.min_heap):
            return (-self.max_heap[0] + self.min_heap[0]) / 2
        # 否则中位数为堆顶元素较多的那个堆的堆顶
        elif len(self.max_heap) > len(self.min_heap):
            return -self.max_heap[0]
        else:
            return self.min_heap[0]

if __name__ == '__main__':
    median = DynamicMedian()
    nums = [5, 2, 3, 4, 1, 6, 7, 0, 8]
    for num in nums:
        median.insert(num)
        print('当前中位数为：', median.find_median())

当前中位数为： 5
当前中位数为： 3.5
当前中位数为： 3
当前中位数为： 3.5
当前中位数为： 3
当前中位数为： 3.5
当前中位数为： 4
当前中位数为： 3.5
当前中位数为： 4