bisect模块

bisect模块包含两个主要函数，bisect和insort，两个函数都利用二分查找算法来在有序的序列中查找或插入元素。

用bisect搜索

在有序序列中用bisect 查找某个元素的插入位置：

import bisect
import sys
HAYSTACK = [1, 4, 5, 6, 8, 12, 15, 20, 21, 23, 23, 26, 29, 30]

NEEDLES = [0, 1, 2, 5, 8, 10, 22, 23, 29, 30, 31]

ROW_FMT = '{0:2d} @ {1:2d} {2}{0:<2d}'

def demo(bisect_fn):

for needle in reversed(NEEDLES):

#用特定的bisect函数来计算元素应该出现的位置

position = bisect_fn(HAYSTACK,needle)

offset = position * '  |'

# 把元素和其应该出现的位置打印出来

print(ROW_FMT.format(needle, position, offset))
if name == 'main':

if sys.argv[-1] == 'left':

bisect_fn = bisect.bisect_left

else:

bisect_fn = bisect.bisect
print(&#39;DEMO:&#39;, bisect_fn.__name__)  #把选定的函数打印出来
print(&#39;haystack -&gt;&#39;, &#39; &#39;.join(&#39;%2d&#39; % n for n in HAYSTACK))
demo(bisect_fn)

 

实际应用

import bisect
根据一个分数，找到对应的成绩
def grade(score,breakpoints = [60,70,80,90], grades="FDCBA"):

#找到分数在breakpoints中的位置

i = bisect.bisect(breakpoints,score)

#用该位置找到grade中对应的成绩，并返回

return grades[i]
print([grade(score) for score in [22,63,75,12,98]])
['F', 'D', 'C', 'F', 'A']

 

 

用bisect.insort插入新元素

排序很耗时，因此在得到一个有序序列之后，我们最好能够保持它的有序。bisect.insort 就是为了这个而存在的。

insort(seq, item)

把变量item 插入到序列seq 中，并能保持seq 的升序顺序。

# insort 可以保持有序序列的顺序
import bisect
import random
SIZE = 8
seed() 方法改变随机数生成器的种子,使random() 生成的随机数相同
random.seed(7)
原有列表（有序）
my_list = []

for i in range(SIZE):

#randrange()函数在一定的范围内产生随机数

new_item = random.randrange(SIZE*2)

# 在列表中插入元素，并保持列表有序

bisect.insort(my_list,new_item)

print('%2d -> '% new_item,my_list)