forの高速化

Pythonの話じゃなくてPyPyの話かもしれない

2次元配列

forの順番が交換可能なら、添字の順番を外→内にすると速い

forが交換不可能なら、forに併せて合わせて添字が外→内になるように配列を定義したい

後からまとめてindexをつけて参照するよりも、forの直後に配列の参照を変数にブチ込んだほうが速い

遅い

code:matrix_mul_slow.py

for i in range(N):

for j in range(N):

for k in range(N):

若干速い

code:matrix_mul_slow.py

for i in range(N):

for k in range(N):

for j in range(N):

速い

code:matrix_mul_fast.py

for i in range(N):

for k in range(N):

for j in range(N):

内側の配列の大きさが大きくなるように設計しても速いらしい

numpy や numba で高速化させる方法もあるらしい

多重for

配列から何個か取り出す組み合わせ全探索を多重forで作る時、添字小さい方から決める感じにした方が速い

速い

code:combination_fast.py

for i1 in range(N):

for i2 in range(i1+1, N):

for i3 in range(i2+1, N):

for i4 in range(i3+1, N):

for i5 in range(i4+1, N):

hogehoge()

遅い

code:combination_slow.py

for i1 in range(N):

for i2 in range(i1):

for i3 in range(i2):

for i4 in range(i3):

for i5 in range(i4):

hogehoge()

code:combination_itertools.py

from itertools import combinations

for b in combinations(range(N), 5):

hogehoge()

Pythonだと速度差はあまりない