在Python中定时函数

时间:2013-01-24 19:07:29

标签: python search recursion time

我正在尝试在python中计算两个不同的函数。

第一个:

import cProfile
def bin_search(A, first,last, target):
#returns index of target in A, if present
#returns -1 if target is not present in A
if first > last:
    return -1
else:
    mid = (first+last)/2
    if A[mid]==target:
        return mid
    elif A[mid]>target:
        return bin_search(A,first,mid-1,target)
    else:
        return bin_search(A,mid+1,last,target)

第二个

def trin_search(A,first,last,target):
#returns index of target in A, if present
#returns -1 if target is not present in A
if target> last or target<first:
    return -1
if first>last:
    return -1
else:
    one_third=first+(last-first)/3
    two_thirds=first+2*(last-first)/3
    if A[one_third]==target:
        return one_third
    elif A[one_third]>target:
        #search the left-hand third
        return trin_search(A,first, one_third,target)
    elif A[two_thirds]==target:
        return two_thirds
    elif A[two_thirds]>target:
        #search the middle third
        return trin_search(A,one_third+1,two_thirds-1,target)
    else:
        #search the right-hand third
        return trin_search(A,two_thirds+1,last,target)

我正在尝试使用cprofile.run()方法计时。我打电话给:

cprofile.run('trin_search(newlist, newlist[0], newlist[-1], 17)')

cprofile.run('bin_search(newlist, newlist[0], newlist[-1], 17)')

第一个结果:

6 function calls (4 primitive calls) in 0.000 seconds

Ordered by: standard name

ncalls  tottime  percall  cumtime  percall filename:lineno(function)
    1    0.000    0.000    0.000    0.000 :0(setprofile)
    1    0.000    0.000    0.000    0.000 <string>:1(<module>)
  3/1    0.000    0.000    0.000    0.000 Jan 18.py:16(trin_search)
    0    0.000             0.000          profile:0(profiler)
    1    0.000    0.000    0.000    0.000 profile:0(trin_search(newlist, newlist[0], newlist[-1], 17))

和第二个

7 function calls (3 primitive calls) in 0.000 seconds

Ordered by: standard name

ncalls  tottime  percall  cumtime  percall filename:lineno(function)
    1    0.000    0.000    0.000    0.000 <string>:1(<module>)
  5/1    0.000    0.000    0.000    0.000 Jan 18.py:2(bin_search)
    1    0.000    0.000    0.000    0.000 {method 'disable' of '_lsprof.Profiler' objects}    

他们怎么可能花0时间来操作?

干杯,

2 个答案:

答案 0 :(得分:3)

正如其他人使用timeit模块所指出的,下面是一个如何使用参数计算函数的示例:

import timeit

arg = 10

def foo(arg):
    return arg**arg

t=timeit.Timer("foo(arg)","from __main__ import foo, arg")
print t.timeit(5)

请注意,您必须导入函数调用中使用的函数和变量。

此外,我建议您使用IPython “magic commands”,这样您就可以%timeit foo(arg)


对于您的示例,这应该有效:

t=timeit.Timer("bin_search(newlist, newlist[0], newlist[-1], 17)",
                       "from __main__ import bin_search, newlist") 

答案 1 :(得分:0)

尝试timeit模块。它用于对代码段进行基准测试。