このパッケージは処理時間の計測module timeit を使いやすくしたものです。以下の特徴があります。
- 引数を使い回しません(timeitは引数を書き換える処理をした時にそのまま次に渡します)
- 複数の関数を1度に計測して相対的な速度差を視覚化できます
- 処理時間の桁数を変えられます(0.00374567より0.0037の方が見やすい)
- 返り値を一緒にprintできます
$ pip install timeit2
from timeit2 import ti2
arg = range(10 ** 6)
ti2(
max,
sum,
args=[arg],
floor=4,
print_return=True,
return_label="ret: ",
relative=True,
)
# max:
# 0.0356 sec
# ret: 999999
# sum:
# 0.0249 sec
# ret: 499999500000
# relative:
# sum:
# 1
# max:
# 1.43最小限の使い方
from timeit2 import ti2
def f():
li = []
for i in range(10 ** 5):
li.append(i * i)
return li
ti2(f)
# f:
# 0.015094 sec配列作成にlist.append,list.insert,dequeをそれぞれ使った処理時間と相対評価
from collections import deque
from timeit2 import ti2
def append_(n):
li = []
for i in range(n):
li.append(i)
return li
def insert_(n):
li = []
for i in range(n):
li.insert(0, i)
return li
def deque_right(n):
li = deque()
for i in range(n):
li.append(i)
return li
def deque_left(n):
li = deque()
for i in range(n):
li.appendleft(i)
return li
ti2(
append_,
insert_,
deque_right,
deque_left,
args=[10 ** 4],
relative=True,
)
# append_:
# 0.001118 sec
# insert_:
# 0.021187 sec
# deque_right:
# 0.000891 sec
# deque_left:
# 0.000855 sec
# relative:
# deque_left:
# 1
# deque_right:
# 1.04
# append_:
# 1.31
# insert_:
# 24.77ti2(
*fn: Callable,
args: list = [],
number: int = 100,
floor: int = 6,
relative: bool = False,
print_return: bool = False,
return_label: str = "",
)Type: Callable
計測する関数、可変長引数なので複数指定が可能。
引数がある場合はargs= で指定する。
Type: list
Default: []
fnに渡す引数のlist
Type: int
Default: 100
各関数を計測する回数、結果はその最速値が表示される
Type: int
Default: 6
表示する小数点以下の桁数
Type: bool
Default: False
計測結果に相対評価を追加する
Type: bool
Default: False
計測結果に関数の返り値を追加する
Type: str Default: ""
print_return=Trueの場合、返り値のラベルを指定する
ti2, ti, timeit, timeit2 をimportできますが、全て同じものです。
from timeit2 import ti2, ti, timeit, timeit2MIT © 2021 atu4403