代码之家 › 专栏 › 技术社区 › Rodolphe Lemasquerier

在python中使用heapq获取优先级列表的问题

priority-queue heap object list python

Rodolphe Lemasquerier · 技术社区 · 8 年前

我不明白为什么我下面的代码会出错。

我正在尝试基于Python的heapq模块构建一个优先级列表。该模块的一个基本示例的唯一区别是希望将其用于其中的自定义对象,而不是简单的(int,int)或(int,str)元组。

import heapq

class MyObject():

 def __init__(self,a=0,name='toto'):

     self.a = a
     self.name = name

if __name__ == '__main__':

 priority_list = []
 heapq.heappush(priority_list,(1,MyObject()))
 heapq.heappush(priority_list,(1,MyObject()))

这是我的错误:

heapq.heappush(priority_list,(1,MyObject()))

TypeError: '<' not supported between instances of 'MyObject' and 'MyObject'

如果使用不同的键插入堆中,则不会引发错误,但heapq不应该处理相同的键吗? 我不太理解这种行为。

非常感谢

2 回复 | 直到 8 年前

Pedro H. N. Vieira 8 年前

操作员 < 未为您的类定义。那样的话 heapq 无法定义优先级。

ob1 = MyObject()
ob1 < ob1

提高

TypeError: unorderable types: MyObject() < MyObject()

然后必须定义逻辑运算符。看见 this 了解更多信息。

class MyObject():
    def __init__(self,a=0,name='toto'):
        self.a = a
        self.name = name

    def __lt__(ob1, ob2):
        return ob1.a < ob2.a

ob1 = MyObject()
ob1 < ob1 # returns False

Daniel Roseman 8 年前

堆的属性是最小的对象始终位于顶部。为了让Python保持不变,它必须有某种方法来确定哪个对象更小。MyObject类不提供此功能。

您可以定义 __gt__ 或 __lt__ 以启用此功能。

推荐文章

Reegan · 使用Angular ngFor指令在html中渲染嵌套对象值

1 年前

ShaAnder · 为什么sqllachemy返回的是类而不是字符串

1 年前

D_chez12 · 如果对象键值为空,则过滤掉数组中的对象

1 年前

Mohit · React useContext():当试图在useContext(()api中执行函数以获取数据时,对象作为React子对象无效

1 年前

Powl_London · Javascript Object,返回一个新数组,其中包含与该对象匹配的索引[重复]

1 年前

MeltingDog · 如何检索嵌套的JS对象并通过索引获取其数据?

1 年前

London28 · Javascript,映射一个对象数组以返回新数组[重复]

1 年前

bolthy · 将数据帧中的值从对象转换为整数/浮点数,但它们仍作为对象运行

1 年前

SamCharles · 在JavaScript的对象数组中查找对象数组中不区分大小写的子字符串

1 年前

VonDerHase · 从列表中删除特定值,Python

1 年前