前言
bisect实现了一个算法来向列表中插入元素,同时仍保持列表有序。
本篇,将详细介绍bisect库高效率的玩转列表。
有序插入
首先,我们来看看bisect库是如何实现列表的插入的。具体代码如下所示:
- import bisect
- a = [7, 5, 4, 1, 9, 8, 2, 3, 6, 0, 5]
- print(a)
- new_a = []
- for i in a:
- position = bisect.bisect(new_a, i)
- bisect.insort(new_a, i)
- print(position, new_a)
运行之后,效果如下:
可以看到,bisect会自动排序进行插入,position为插入的索引位置。当然,对于此类插入如果直接构建列表,然后排序,可能速度更快。不过这只仅仅对于短列表而言非常的快,对于非常长的列表而言,使用上面这种插入排序方式可以大大节省时间和内存,尤其是比较两个列表成员的操作需要开销很大的计算量时。
重复值处理
在实际的列表处理中,我们可能处理重复的值。如前文所示,多余的5是默认插入到重复值右边的,也就是说相当于使用insort_right()函数。同理,那么左边我们就可以用insort_left()函数。
- import bisect
- a = [7, 5, 4, 1, 9, 8, 2, 3, 6, 0, 5]
- print(a)
- new_a = []
- for i in a:
- position = bisect.bisect_left(new_a, i)
- bisect.insort_left(new_a, i)
- print(position, new_a)
运行之后,效果如下:
读者可以对比一下上面两个图片,最后一行的索引变化。可以看到,一个是6,一个是5,因为我们主动变更,把重复值默认插入到左边了。