Python3标准库：bisect维护有序列表

Posted 2021-03-08 爱编程的小灰灰

1. bisect维护有序列表

bisect模块实现了一个算法来向列表中插入元素，同时仍保持列表有序。

1.1 有序插入

下面给出一个简单的例子，这里使用insort()按有序顺序向一个列表中插入元素。

import bisect

# A series of random numbers
values = [14, 85, 77, 26, 50, 45, 66, 79, 10, 3, 84, 77, 1]

print(\'New  Pos  Contents\')
print(\'---  ---  --------\')

l = []
for i in values:
    position = bisect.bisect(l, i)
    bisect.insort(l, i)
    print(\'{:3}  {:3}\'.format(i, position), l)

输出的第一列显示了新的随机数。第二轮显示了这个数将插入到列表的哪个位置。每一行余下的部分则是当前的有序列表。

这是一个很简单的例子，实际上，对于此例处理的数据量来说，如果直接构建列表然后完成一次排序，可能速度更快。不过对于长列表而言，使用类似这样的一个插入排序算法可以大大节省时间和内存，尤其是比较两个列表成员的操作需要开销很大的计算时。 

1.2 处理重复

之前显示的结果集包括一个重复的值77。bisect模块提供了两种方法来处理重复。新值可以插入到原值的左边或右边。insort()函数实际上是insort_right()的别名，这个函数会在原值之后插入新值。相应的insort_left()函数则在原值之前插入新值。

import bisect

# A series of random numbers
values = [14, 85, 77, 26, 50, 45, 66, 79, 10, 3, 84, 77, 1]

print(\'New  Pos  Contents\')
print(\'---  ---  --------\')

# Use bisect_left and insort_left.
l = []
for i in values:
    position = bisect.bisect_left(l, i)
    bisect.insort_left(l, i)
    print(\'{:3}  {:3}\'.format(i, position), l)

使用bisect_left()和insort_left()处理同样的数据时，结果是相同的有序列表，不过重复值插入的位置有所不同。