Python 的列表(list)内部实现是一个数组,也就是一个线性表。在列表中查找元素可以使用 list.index() 方法,其时间复杂度为O(n)。对于大数据量,则可以用二分查找进行优化。二分查找要求对象必须有序,其基本原理如下:

1.从数组的中间元素开始,如果中间元素正好是要查找的元素,则搜素过程结束;

2.如果某一特定元素大于或者小于中间元素,则在数组大于或小于中间元素的那一半中查找,而且跟开始一样从中间元素开始比较。

3.如果在某一步骤数组为空,则代表找不到。

二分查找也成为折半查找,算法每一次比较都使搜索范围缩小一半, 其时间复杂度为 O(logn)。

我们分别用递归和循环来实现二分查找:

def binary_search_recursion(lst, value, low, high):

if high < low:

return None

mid = (low + high) / 2

if lst[mid] > value:

return binary_search_recursion(lst, value, low, mid-1)

elif lst[mid] < value:

return binary_search_recursion(lst, value, mid+1, high)

else:

return mid

def binary_search_loop(lst,value):

low, high = 0, len(lst)-1

while low <= high:

mid = (low + high) / 2

if lst[mid] < value:

low = mid + 1

elif lst[mid] > value:

high = mid - 1

else:

return mid

return None

接着对这两种实现进行一下性能测试:

if __name__ == "__main__":

import random

lst = [random.randint(0, 10000) for _ in xrange(100000)]

lst.sort()

def test_recursion():

binary_search_recursion(lst, 999, 0, len(lst)-1)

def test_loop():

binary_search_loop(lst, 999)

import timeit

t1 = timeit.Timer("test_recursion()", setup="from __main__ import test_recursion")

t2 = timeit.Timer("test_loop()", setup="from __main__ import test_loop")

print "Recursion:", t1.timeit()

print "Loop:", t2.timeit()

执行结果如下:

Recursion: 3.12596702576

Loop: 2.08254289627

可以看出循环方式比递归效率高。

Python 有一个 bisect 模块,用于维护有序列表。bisect 模块实现了一个算法用于插入元素到有序列表。在一些情况下,这比反复排序列表或构造一个大的列表再排序的效率更高。Bisect 是二分法的意思,这里使用二分法来排序,它会将一个元素插入到一个有序列表的合适位置,这使得不需要每次调用 sort 的方式维护有序列表。

下面是一个简单的使用示例:

import bisect

import random

random.seed(1)

print'New Pos Contents'

print'--- --- --------'

l = []

for i in range(1, 15):

r = random.randint(1, 100)

position = bisect.bisect(l, r)

bisect.insort(l, r)

print'%3d %3d' % (r, position), l

输出结果:

New Pos Contents

--- --- --------

14 0 [14]

85 1 [14, 85]

77 1 [14, 77, 85]

26 1 [14, 26, 77, 85]

50 2 [14, 26, 50, 77, 85]

45 2 [14, 26, 45, 50, 77, 85]

66 4 [14, 26, 45, 50, 66, 77, 85]

79 6 [14, 26, 45, 50, 66, 77, 79, 85]

10 0 [10, 14, 26, 45, 50, 66, 77, 79, 85]

3 0 [3, 10, 14, 26, 45, 50, 66, 77, 79, 85]

84 9 [3, 10, 14, 26, 45, 50, 66, 77, 79, 84, 85]

44 4 [3, 10, 14, 26, 44, 45, 50, 66, 77, 79, 84, 85]

77 9 [3, 10, 14, 26, 44, 45, 50, 66, 77, 77, 79, 84, 85]

1 0 [1, 3, 10, 14, 26, 44, 45, 50, 66, 77, 77, 79, 84, 85]

Bisect模块提供的函数有:

bisect.bisect_left(a,x, lo=0, hi=len(a)) :

查找在有序列表 a 中插入 x 的index。lo 和 hi 用于指定列表的区间,默认是使用整个列表。如果 x 已经存在,在其左边插入。返回值为 index。

bisect.bisect_right(a,x, lo=0, hi=len(a))

bisect.bisect(a, x,lo=0, hi=len(a)) :

这2个函数和 bisect_left 类似,但如果 x 已经存在,在其右边插入。

bisect.insort_left(a,x, lo=0, hi=len(a)) :

在有序列表 a 中插入 x。和 a.insert(bisect.bisect_left(a,x, lo, hi), x) 的效果相同。

bisect.insort_right(a,x, lo=0, hi=len(a))

bisect.insort(a, x,lo=0, hi=len(a)) :

和 insort_left 类似,但如果 x 已经存在,在其右边插入。

Bisect 模块提供的函数可以分两类: bisect* 只用于查找 index, 不进行实际的插入;而 insort* 则用于实际插入。该模块比较典型的应用是计算分数等级:

def grade(score,breakpoints=[60, 70, 80, 90], grades='FDCBA'):

i = bisect.bisect(breakpoints, score)

return grades[i]

print [grade(score) for score in [33, 99, 77, 70, 89, 90, 100]]

执行结果:

['F', 'A', 'C', 'C', 'B', 'A', 'A']

同样,我们可以用 bisect 模块实现二分查找:

def binary_search_bisect(lst, x):

from bisect import bisect_left

i = bisect_left(lst, x)

if i != len(lst) and lst[i] == x:

return i

return None

我们再来测试一下它与递归和循环实现的二分查找的性能:

Recursion: 4.00940990448

Loop: 2.6583480835

Bisect: 1.74922895432

可以看到其比循环实现略快,比递归实现差不多要快一半。

Python 著名的数据处理库 numpy 也有一个用于二分查找的函数 numpy.searchsorted, 用法与 bisect 基本相同,只不过如果要右边插入时,需要设置参数 side='right',例如:

>>> import numpy as np

>>> from bisect import bisect_left, bisect_right

>>> data = [2, 4, 7, 9]

>>> bisect_left(data, 4)

1

>>> np.searchsorted(data, 4)

1

>>> bisect_right(data, 4)

2

>>> np.searchsorted(data, 4, side='right')

2

那么,我们再来比较一下性能:

In [20]: %timeit -n 100 bisect_left(data, 99999)

100 loops, best of 3: 670 ns per loop

In [21]: %timeit -n 100 np.searchsorted(data, 99999)

100 loops, best of 3: 56.9 ms per loop

In [22]: %timeit -n 100 bisect_left(data, 8888)

100 loops, best of 3: 961 ns per loop

In [23]: %timeit -n 100 np.searchsorted(data, 8888)

100 loops, best of 3: 57.6 ms per loop

In [24]: %timeit -n 100 bisect_left(data, 777777)

100 loops, best of 3: 670 ns per loop

In [25]: %timeit -n 100 np.searchsorted(data, 777777)

100 loops, best of 3: 58.4 ms per loop

可以发现 numpy.searchsorted 效率是很低的,跟 bisect 根本不在一个数量级上。因此 searchsorted 不适合用于搜索普通的数组,但是它用来搜索 numpy.ndarray 是相当快的:

In [30]: data_ndarray = np.arange(0, 1000000)

In [31]: %timeit np.searchsorted(data_ndarray, 99999)

The slowest run took 16.04 times longer than the fastest. This could mean that an intermediate result is being cached.

1000000 loops, best of 3: 996 ns per loop

In [32]: %timeit np.searchsorted(data_ndarray, 8888)

The slowest run took 18.22 times longer than the fastest. This could mean that an intermediate result is being cached.

1000000 loops, best of 3: 994 ns per loop

In [33]: %timeit np.searchsorted(data_ndarray, 777777)

The slowest run took 31.32 times longer than the fastest. This could mean that an intermediate result is being cached.

1000000 loops, best of 3: 990 ns per loop

numpy.searchsorted 可以同时搜索多个值:

>>> np.searchsorted([1,2,3,4,5], 3)

2

>>> np.searchsorted([1,2,3,4,5], 3, side='right')

3

>>> np.searchsorted([1,2,3,4,5], [-10, 10, 2, 3])

array([0, 5, 1, 2])

python searchsorted_Python 二分查找与 bisect 模块相关推荐

  1. python bisect_Python实现二分查找与bisect模块详解

    前言 其实Python 的列表(list)内部实现是一个数组,也就是一个线性表.在列表中查找元素可以使用 list.index()方法,其时间复杂度为O(n) .对于大数据量,则可以用二分查找进行优化 ...

  2. Python二分查找与插入模块bisect

    Python的二分查找与插入bisect模块,查询函数主要使用bisect.bisect(),bisect.bisect_left(),bisect.bisect_right(),插入函数主要使用bi ...

  3. python实现二分查找(折半查找)算法

    python实现二分查找算法 二分查找也称折半查找(Binary Search),它是一种效率较高的查找方法.但是,折半查找要求线性表必须采用顺序存储结构,而且表中元素按关键字有序排列. 查找又称折半 ...

  4. python中二分查找什么意思_python中二分查找法的实现方法

    如果想要在有序数据中进行查找想要的数据,二分查找法就个好方法,它可以大大缩短了搜索时间,是一种常见的查找方法.二分查找很好写,却很难写对,下面,小编就简单向大家介绍一下二分查找,并演示器使用代码. 1 ...

  5. 七十六、Python | Leetcode二分查找和分治算法系列

    @Author:Runsen @Date:2020/7/4 人生最重要的不是所站的位置,而是内心所朝的方向.只要我在每篇博文中写得自己体会,修炼身心:在每天的不断重复学习中,耐住寂寞,练就真功,不畏艰 ...

  6. python二分查找算法_如何使用python的二分查找算法

    如何使用python的二分查找算法 发布时间:2020-11-18 09:27:29

  7. python函数教程:Python递归函数 二分查找算法实现解析

    这篇文章主要介绍了Python递归函数 二分查找算法实现解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 一.初始递归 递归函数:在一个函数里在调 ...

  8. python实现二分查找(对新手友好,内容通俗易懂)

    python实现二分查找 二分查找又名折半查找. 优点:查询速度快,性能好. 缺点:要求查询的表为有序表 原理:将表中间位置(mid)的数字与待查数字(data)做比较, 如果相等:返回true,结束 ...

  9. python实现二分查找代码+详解

    python实现二分查找代码+详解 一.规定函数值 函数输入值 li=[1,2,3,4,5,6,7,8,9]#输入的集合 val=4#要查找的值 函数内部值 left=0#代表集合中第一个值的位置 r ...

  10. Python实现二分查找算法,图片结合代码讲解,简单易懂

    Python实现二分查找算法,图片结合代码讲解,简单易懂 二分查找是指在一个数组中循环找到特定的目标元素的一种算法,二分查找算法要求这个数组必须是有序的,如果不是有序则最后得出的结果是不准确的,因为其 ...

最新文章

  1. 【深度学习】循环神经网络(RNN)简易教程
  2. mysql单表最大数据量_你的Mysql库真需要Adaptive Hash Index
  3. C++学习之CodeBlocks安装与调试
  4. tomcat出现5个using_下肢深静脉血栓要警惕,出现这5个症状,马上看医生
  5. 【clickhouse】Too many parts . Merges are processing significantly slower than inserts
  6. 用cxf编写基于spring的webservice之上篇
  7. 【more effective c++读书笔记】【第5章】技术(5)——Reference counting(引用计数)(2)...
  8. 2014恒生电子笔试题
  9. [html] 网页打印与标准纸张换算时,cm和px是如何换算的?
  10. 一文掌握机器学习算法工程师技术栈
  11. ue4 迁移模型_UE4模型导入基础操作(MAX为例)
  12. matlab写子函数,matlab主函数子函数怎么写
  13. 泊松过程的概念及其例题分析
  14. idea报错:fatal: –author ‘user@mail.com’ is not ‘Name ’ and matches no existing author
  15. 关于使用https请求微信乱码问题
  16. 计算机运维方向要考什么证,IT运维项目经理考的证
  17. 1913. 公平摄影
  18. LC117 Populating Next Right Pointers in Each Node II
  19. 斗鱼直播行业名列前茅 泛娱乐布局成效凸显
  20. C++ 主函数几种语法

热门文章

  1. 概念理解之装箱与拆箱
  2. 原创 leetcode[454]四数相加II /4Sum II 哈希策略
  3. Lightroom 教程,如何将照片从 Lightroom 移至Photoshop,在 Ps 中合并图像?
  4. Ps 初学者教程,如何用文字增强您的照片?
  5. 如何在 iPhone 和 Mac 上的 Apple Notes 中使用智能文件夹?
  6. Image2icon for Mac(icon图标设计软件)
  7. HTTPS客户端验证服务器端的过程
  8. idea 配置 maven 和 镜像
  9. 【Codeforces】868C. Qualification Rounds
  10. MSP430学习小结2-程序主体结构安排及低功耗-转载