python - 在排序列表中插入自定义对象

Question

我实现了一个带有数字属性的对象。我想保留根据该属性排序的这些对象的列表，而无需在每次插入时运行排序方法。我看了一下 bisect 模块，但我不知道我是否也可以将它与对象一起使用。
最好的方法是什么？

Answer 1

如果您实现 __lt__ ，您可以为您的自定义对象执行此操作。方法，因为 this is what bisect will use比较你的对象。

>>> class Foo(object):
...     def __init__(self, val):
...         self.prop = val # The value to compare
...     def __lt__(self, other):
...         return self.prop < other.prop
...     def __repr__(self):
...         return 'Foo({})'.format(self.prop)
...
>>> sorted_foos = sorted([Foo(7), Foo(1), Foo(3), Foo(9)])
>>> sorted_foos
[Foo(1), Foo(3), Foo(7), Foo(9)]
>>> bisect.insort_left(sorted_foos, Foo(2))
>>> sorted_foos
[Foo(1), Foo(2), Foo(3), Foo(7), Foo(9)]