如何按指定索引将矩阵内列表推入二叉堆并保留原列表？报错求助

嗨，我来帮你理清这个问题的解决思路：

首先解决你遇到的报错：TypeError: heappush() takes no keyword arguments #

你之前写的heapq.heappush(_heap, item=itemgetter(-1))有两个核心问题：

1. heapq.heappush的参数是位置参数，不支持用item=这种关键字传参的方式，正确的传参格式是heappush(堆对象, 要推入的元素)。
2. itemgetter(-1)是一个用于提取元素的函数，不是要推入堆的实际元素——你需要把它作用在具体的子列表上，比如itemgetter(-1)(子列表)才能拿到用于排序的键值。

按指定索引将子列表推入堆并保留原列表的通用方法 #

要实现「按子列表的指定索引排序，同时保留原列表」的需求，最简洁通用的方式是推入包含「排序键值+原列表」的元组，因为Python的heapq模块默认会按元组的第一个元素进行排序，排序完成后我们依然可以取出完整的原列表。

代码示例（针对你的场景） #

import heapq
from operator import itemgetter

l1 = [[1, 3], [3, 2], [2, 1]]
memory = []

# 遍历每个子列表，推入(排序键值, 原列表)到堆中
for item in l1:
    # 这里用item[-1]直接取最后一个元素，或者用itemgetter(-1)(item)效果完全一致
    heapq.heappush(memory, (item[-1], item))

# 验证堆的弹出顺序（按排序键值从小到大）
while memory:
    sort_key, original_item = heapq.heappop(memory)
    print(f"弹出的排序键: {sort_key}, 原列表: {original_item}")

运行这段代码会输出：

弹出的排序键: 1, 原列表: [2, 1]
弹出的排序键: 2, 原列表: [3, 2]
弹出的排序键: 3, 原列表: [1, 3]

通用化扩展（支持任意指定索引） #

如果以后需要按其他索引（比如索引0）排序，只需要把item[-1]换成item[k]（k是你指定的索引）即可，还可以封装成通用函数：

import heapq

def push_to_heap(heap, matrix, sort_index):
    for item in matrix:
        heapq.heappush(heap, (item[sort_index], item))

# 调用示例：按索引0排序
memory = []
push_to_heap(memory, l1, sort_index=0)

备选方案：自定义类实现复杂排序规则 #

如果你的场景更复杂（比如需要多条件排序、反向排序等），可以通过自定义类并实现__lt__方法来指定排序逻辑：

import heapq

class HeapElement:
    def __init__(self, data, sort_index):
        self.data = data
        self.sort_index = sort_index
    
    def __lt__(self, other):
        # 按指定索引的元素进行比较，这里可以扩展任意排序逻辑
        return self.data[self.sort_index] < other.data[self.sort_index]

l1 = [[1, 3], [3, 2], [2, 1]]
memory = []

for item in l1:
    heapq.heappush(memory, HeapElement(item, sort_index=-1))

# 弹出时获取原列表
while memory:
    elem = heapq.heappop(memory)
    print(f"原列表: {elem.data}")

这个方案虽然代码量多一点，但灵活性更高，适合非简单键值排序的场景。

内容的提问来源于stack exchange，提问作者Ross Nkama