テクめも

プログラミング関連のちょっとしたTipsなどを書いています。

Pythonで多重ループを1行にまとめる

Pythonで多重ループを書くとき、標準ライブラリのitertools.productを使うと便利に書けます。

例えば、以下のような3重ループも

# A, B, Cはリスト
for i in A:
    for j in B:
        for k in C:
            print(i, j, k)

このように、1行で書くことができます。

from itertools import product
for i, j, k in product(A, B, C):
    print(i, j, k)