如何获取2-dim数值数组对的topK项？

import heapq

def get_top_k_pairs(arr, k):
    # 创建一个最小堆
    heap = []
    # 遍历2-dim数组，将二元组存储在最小堆中
    for i, row in enumerate(arr):
        for j, val in enumerate(row):
            # 使用负数将最小堆转换为最大堆
            heapq.heappush(heap, (-val, (i, j)))
            # 保持堆的大小为k
            if len(heap) > k:
                heapq.heappop(heap)
    
    # 取出堆中的所有元素
    top_k_pairs = [pair for _, pair in heap]
    # 将结果按照数值大小进行升序排序
    top_k_pairs.sort(key=lambda x: x[0], reverse=True)
    
    return top_k_pairs

# 测试
arr = [
    [5, 7, 3],
    [2, 8, 4],
    [1, 6, 9]
]
k = 2
top_k_pairs = get_top_k_pairs(arr, k)
print(top_k_pairs)