如何在Python中计算一批代码的执行时间?

时间:2012-01-17 02:47:18

标签: python time

我在StackOverflow上找到了这个问题和答案。

Python - time.clock() vs. time.time() - accuracy?

以下是我正在尝试运行的一些代码:

import sys
import time
import timeit

if (len(sys.argv) > 1):
  folder_path = sys.argv[1]
  if not os.path.isdir(folder_path):
    print "The folder you provided doesn't exist"    
  else:
    print_console_headers()    
    rename_files_to_title_case(folder_path)
    #start the timer.
    #do some freaky magic here.
    #end the timer.

else:
  print "You must provide a path to a folder."

def print_console_headers():
  print "Renaming files..."
  print "--------------------"
  return

def rename_files_to_title_case():  
  """this is just for testing purposes"""  
  L = []
  for i in range(100):
      L.append(i)

if __name__ == '__main__':
    from timeit import Timer
    t = Timer("test()", "from __main__ import test")
    print t.timeit()

我如何使用已保存在别处的参数给timeit一个函数?

这是我用Ruby编写的,它给了我干净代码的结果,也许这有助于提出建议。

start_time = Time.now

folder_path = ARGV[0]
i = 0
Dir.glob(folder_path + "/*").sort.each do |f|
    filename = File.basename(f, File.extname(f))
    File.rename(f, folder_path + "/" + filename.gsub(/\b\w/){$&.upcase} + File.extname(f))
    i += 1
end

puts "Renaming complete."
puts "The script renamed #{i} file(s) correctly."
puts "----------"
puts "Running time is #{Time.now - start_time} seconds"

4 个答案:

答案 0 :(得分:34)

这就是我通常用Python编写时间测量代码的方法:

start_time = time.time()

# ... do stuff

end_time = time.time()
print("Elapsed time was %g seconds" % (end_time - start_time))

正如您链接的帖子所述,time.clock()不适合测量已用时间,因为它仅报告您的进程使用的CPU时间量(至少在Unix系统上) )。使用time.time()是跨平台且可靠的。

答案 1 :(得分:14)

我会使用时序装饰器并将您想要的代码放入函数中。

import time

def timeit(f):

    def timed(*args, **kw):

        ts = time.time()
        result = f(*args, **kw)
        te = time.time()

        print 'func:%r args:[%r, %r] took: %2.4f sec' % \
          (f.__name__, args, kw, te-ts)
        return result

    return timed

使用装饰器很容易使用注释。

@timeit
def compute_magic(n):
     #function definition
     #....

或者重新设置您想要的时间功能。

compute_magic = timeit(compute_magic)

我的博文在这里有更多信息。 http://blog.mattalcock.com/2013/2/24/timing-python-code/

答案 2 :(得分:4)

时间函数的一种有趣方式是使用装饰器和包装器函数。我用它的一个功能是:

import time

def print_timing(func):
    def wrapper(*arg):
        t1 = time.time()
        res = func(*arg)
        t2 = time.time()
        string = '| %s took %0.3f ms |' % (func.func_name, (t2-t1)*1000.0)
        print
        print '-'*len(string)
        print string
        print '-'*len(string)
        print
        return res
    return wrapper

由@print_timing修饰的任何函数都将打印打印到stdout所用的时间

@print_timing
def some_function(text):
    print text

这使得计时特定功能变得非常方便。

答案 3 :(得分:2)

如果你需要测量函数中特定代码行的时间,那么当装饰者无法帮助解决问题时

import time
from time import sleep

class TT(object):
    def __init__(self):
        self.start = time.time()
    def __str__(self):
        return str(time.time() - self.start)


timeit = TT()

sleep(1)
print timeit

sleep(2)
print timeit

将打印:

1.0
3.00