Avec l'aide d'une petite classe de commodité, vous pouvez mesurer le temps passé dans les lignes en retrait comme ça :
with CodeTimer():
line_to_measure()
another_line()
# etc...
Qui affichera ce qui suit après que la ou les lignes indentées aient fini de s'exécuter :
Code block took: x.xxx ms
UPDATE : Vous pouvez maintenant obtenir la classe avec pip install linetimer
et ensuite from linetimer import CodeTimer
. Voir ce projet GitHub .
Le code de la classe ci-dessus :
import timeit
class CodeTimer:
def __init__(self, name=None):
self.name = " '" + name + "'" if name else ''
def __enter__(self):
self.start = timeit.default_timer()
def __exit__(self, exc_type, exc_value, traceback):
self.took = (timeit.default_timer() - self.start) * 1000.0
print('Code block' + self.name + ' took: ' + str(self.took) + ' ms')
Vous pourriez alors nommer les blocs de code que vous voulez mesurer :
with CodeTimer('loop 1'):
for i in range(100000):
pass
with CodeTimer('loop 2'):
for i in range(100000):
pass
Code block 'loop 1' took: 4.991 ms
Code block 'loop 2' took: 3.666 ms
Et nid les :
with CodeTimer('Outer'):
for i in range(100000):
pass
with CodeTimer('Inner'):
for i in range(100000):
pass
for i in range(100000):
pass
Code block 'Inner' took: 2.382 ms
Code block 'Outer' took: 10.466 ms
Concernant timeit.default_timer()
il utilise le meilleur timer en fonction du système d'exploitation et de la version de Python, cf. cette réponse .