ฟังก์ชันการวัดเวลา Python

ฉันต้องการสร้างฟังก์ชัน python เพื่อทดสอบเวลาที่ใช้ในแต่ละฟังก์ชันและพิมพ์ชื่อพร้อมเวลาฉันจะพิมพ์ชื่อฟังก์ชันได้อย่างไรและมีวิธีอื่นในการทำอย่างไรโปรดบอกฉันด้วย

def measureTime(a):
    start = time.clock() 
    a()
    elapsed = time.clock()
    elapsed = elapsed - start
    print "Time spent in (function name) is: ", elapsed

แรกและสำคัญที่สุดผมขอแนะนำให้ใช้Profilerหรือการใช้ atleast timeit

อย่างไรก็ตามหากคุณต้องการเขียนวิธีการกำหนดเวลาของคุณเองอย่างเคร่งครัดเพื่อเรียนรู้นี่คือสถานที่ที่จะเริ่มต้นโดยใช้มัณฑนากร

Python 2:

def timing(f):
    def wrap(*args):
        time1 = time.time()
        ret = f(*args)
        time2 = time.time()
        print '%s function took %0.3f ms' % (f.func_name, (time2-time1)*1000.0)
        return ret
    return wrap

และการใช้งานนั้นง่ายมากเพียงใช้ @timing มัณฑนากร:

@timing
def do_work():
  #code

Python 3:

def timing(f):
    def wrap(*args, **kwargs):
        time1 = time.time()
        ret = f(*args, **kwargs)
        time2 = time.time()
        print('{:s} function took {:.3f} ms'.format(f.__name__, (time2-time1)*1000.0))

        return ret
    return wrap

หมายเหตุฉันกำลังเรียกf.func_nameชื่อฟังก์ชันเป็นสตริง (ใน Python 2) หรือf.__name__ ใน Python 3

หลังจากเล่นกับtimeitโมดูลแล้วฉันไม่ชอบอินเทอร์เฟซของมันซึ่งดูไม่ค่อยหรูหราเมื่อเทียบกับสองวิธีต่อไปนี้

โค้ดต่อไปนี้อยู่ใน Python 3

วิธีมัณฑนากร

ซึ่งเกือบจะเหมือนกันกับวิธีการของ @ Mike ที่นี่ฉันเพิ่มkwargsและfunctoolsห่อเพื่อให้ดีขึ้น

def timeit(func):
    @functools.wraps(func)
    def newfunc(*args, **kwargs):
        startTime = time.time()
        func(*args, **kwargs)
        elapsedTime = time.time() - startTime
        print('function [{}] finished in {} ms'.format(
            func.__name__, int(elapsedTime * 1000)))
    return newfunc

@timeit
def foobar():
    mike = Person()
    mike.think(30)

วิธีการจัดการบริบท

from contextlib import contextmanager

@contextmanager
def timeit_context(name):
    startTime = time.time()
    yield
    elapsedTime = time.time() - startTime
    print('[{}] finished in {} ms'.format(name, int(elapsedTime * 1000)))

ตัวอย่างเช่นคุณสามารถใช้เช่น:

with timeit_context('My profiling code'):
    mike = Person()
    mike.think()

และรหัสภายในwithบล็อกจะถูกกำหนดเวลา

ข้อสรุป

ใช้วิธีแรกคุณสามารถแสดงความคิดเห็นกับมัณฑนากรได้ทุกวันเพื่อรับรหัสปกติ อย่างไรก็ตามสามารถจับเวลาได้เฉพาะฟังก์ชันเท่านั้น หากคุณมีโค้ดบางส่วนที่คุณไม่ได้ทำเพื่อให้เป็นฟังก์ชันคุณสามารถเลือกวิธีที่สองได้

ตัวอย่างเช่นตอนนี้คุณมี

images = get_images()
bigImage = ImagePacker.pack(images, width=4096)
drawer.draw(bigImage)

ตอนนี้คุณต้องการเวลาbigImage = ...บรรทัด หากคุณเปลี่ยนเป็นฟังก์ชันจะเป็น:

images = get_images()
bitImage = None
@timeit
def foobar():
    nonlocal bigImage
    bigImage = ImagePacker.pack(images, width=4096)
drawer.draw(bigImage)

ดูไม่ค่อยดีนัก ... จะเป็นอย่างไรถ้าคุณอยู่ใน Python 2 ซึ่งไม่มีnonlocalคีย์เวิร์ด

แทนที่จะใช้วิธีที่สองเหมาะกับที่นี่มาก:

images = get_images()
with timeit_context('foobar'):
    bigImage = ImagePacker.pack(images, width=4096)
drawer.draw(bigImage)

ฉันไม่เห็นว่าtimeitโมดูลมีปัญหาอะไร นี่อาจเป็นวิธีที่ง่ายที่สุดที่จะทำ

import timeit
timeit.timeit(a, number=1)

นอกจากนี้ยังสามารถส่งอาร์กิวเมนต์ไปยังฟังก์ชัน สิ่งที่คุณต้องมีคือรวมฟังก์ชันของคุณโดยใช้มัณฑนากร คำอธิบายเพิ่มเติมที่นี่: http://www.pythoncentral.io/time-a-python-function/

กรณีเดียวที่คุณอาจสนใจในการเขียนคำสั่งเวลาของคุณเองคือถ้าคุณต้องการเรียกใช้ฟังก์ชันเพียงครั้งเดียวและต้องการรับค่าส่งคืนด้วย

ข้อดีของการใช้ timeitโมดูลนี้คือช่วยให้คุณสามารถทำซ้ำจำนวนการดำเนินการได้ สิ่งนี้อาจจำเป็นเนื่องจากกระบวนการอื่น ๆ อาจรบกวนความแม่นยำของเวลาของคุณ ดังนั้นคุณควรเรียกใช้หลาย ๆ ครั้งและดูที่ค่าต่ำสุด

Timeit มีข้อบกพร่องใหญ่สองประการ: จะไม่ส่งคืนค่าส่งคืนของฟังก์ชันและใช้ eval ซึ่งต้องส่งรหัสการตั้งค่าเพิ่มเติมสำหรับการนำเข้า วิธีนี้ช่วยแก้ปัญหาทั้งสองอย่างได้อย่างเรียบง่ายและสวยงาม:

def timed(f):
  start = time.time()
  ret = f()
  elapsed = time.time() - start
  return ret, elapsed

timed(lambda: database.foo.execute('select count(*) from source.apachelog'))
(<sqlalchemy.engine.result.ResultProxy object at 0x7fd6c20fc690>, 4.07547402381897)

มีเครื่องมือที่ง่ายสำหรับการจับเวลา https://github.com/RalphMao/PyTimer

สามารถทำงานได้เหมือนมัณฑนากร :

from pytimer import Timer
@Timer(average=False)      
def matmul(a,b, times=100):
    for i in range(times):
        np.dot(a,b)        

เอาท์พุท:

matmul:0.368434
matmul:2.839355

นอกจากนี้ยังสามารถทำงานเหมือนตัวจับเวลาปลั๊กอินที่มีการควบคุมเนมสเปซ (มีประโยชน์หากคุณกำลังแทรกลงในฟังก์ชันที่มีรหัสจำนวนมากและอาจถูกเรียกใช้ที่อื่น)

timer = Timer()                                           
def any_function():                                       
    timer.start()                                         

    for i in range(10):                                   

        timer.reset()                                     
        np.dot(np.ones((100,1000)), np.zeros((1000,500)))
        timer.checkpoint('block1')                        

        np.dot(np.ones((100,1000)), np.zeros((1000,500)))
        np.dot(np.ones((100,1000)), np.zeros((1000,500)))
        timer.checkpoint('block2')                        
        np.dot(np.ones((100,1000)), np.zeros((1000,1000)))

    for j in range(20):                                   
        np.dot(np.ones((100,1000)), np.zeros((1000,500)))
    timer.summary()                                       

for i in range(2):                                        
    any_function()                                        

เอาท์พุท:

========Timing Summary of Default Timer========
block2:0.065062
block1:0.032529
========Timing Summary of Default Timer========
block2:0.065838
block1:0.032891

หวังว่ามันจะช่วยได้

วิธีการตกแต่งโดยใช้ห้องสมุดมัณฑนากร Python:

import decorator

@decorator
def timing(func, *args, **kwargs):
    '''Function timing wrapper
        Example of using:
        ``@timing()``
    '''

    fn = '%s.%s' % (func.__module__, func.__name__)

    timer = Timer()
    with timer:
        ret = func(*args, **kwargs)

    log.info(u'%s - %0.3f sec' % (fn, timer.duration_in_seconds()))
    return ret

ดูโพสต์ในบล็อกของฉัน:

โพสต์บนบล็อก mobilepro.pl

โพสต์ของฉันบน Google Plus

วิธีการทำของฉัน:

from time import time

def printTime(start):
    end = time()
    duration = end - start
    if duration < 60:
        return "used: " + str(round(duration, 2)) + "s."
    else:
        mins = int(duration / 60)
        secs = round(duration % 60, 2)
        if mins < 60:
            return "used: " + str(mins) + "m " + str(secs) + "s."
        else:
            hours = int(duration / 3600)
            mins = mins % 60
            return "used: " + str(hours) + "h " + str(mins) + "m " + str(secs) + "s."

ตั้งค่าตัวแปรstart = time()ก่อนเรียกใช้ฟังก์ชัน / ลูปและprintTime(start)อยู่หลังบล็อก

และคุณได้รับคำตอบ