In [1]:
import timeit
In [2]:
def test(n):
return sum(range(n))
In [3]:
n = 10000
loop = 1000
In [4]:
result = timeit.timeit('test(n)', globals=globals(), number=loop)
print(result / loop)
In [5]:
result = timeit.timeit(lambda: test(n), number=loop)
print(result / loop)
In [6]:
print(timeit.timeit(lambda: test(n), number=1))
print(timeit.timeit(lambda: test(n), number=10))
print(timeit.timeit(lambda: test(n), number=100))
In [7]:
repeat = 5
print(timeit.repeat(lambda: test(n), repeat=repeat, number=100))
In [8]:
# magic command
# It works only on Jupyter / IPython.
In [9]:
%timeit test(n)
In [10]:
%timeit -r 3 -n 10000 test(n)
In [11]:
%%timeit -r 3 -n 10000
import numpy as np
a = np.arange(n)
np.sum(a)