内包表記の三重ループとif

Question

A = [str(i) + '-' + str(j)  if i != 1 or j == 2 else 'ccc' for i in range(0,2) for j in range(0,3)]

という二重の内包表記を実行すると、結果は

A=['0-0', '0-1', '0-2', 'ccc', 'ccc', '1-2']

となると思います。この内包表記の二重ループを同じように三重ループに拡張して

B=['0-0-0', '0-0-1', '0-0-2', ...,  'ccc', ...]

のような結果にする方法はありますか？内包表記なしのイメージは

B = []
for i in range(0,2):
  for j in range(0,3):
    for k in range(0,4):
      if i != 1 or j == 2:
        B.append(str(i) + '-' + str(j) + '-' + str(k))
      else:
        B.append('ccc')
print(B)

です。三重にした途端に SyntaxError: invalid syntax というエラーが出ました...

Answer 1

多重ループを解消するにはitertools.productを使う手もあります。

from itertools import product
f = lambda i,j,k: str(i) + '-' + str(j) + '-' + str(k) if i != 1 or j == 2 else 'ccc'
C = [f(i,j,k) for i,j,k in product(range(2),range(3),range(4))]
print(C)

Answer 2

以下のコードで試したところ、質問にあるBと同じ出力になりました。

C = [str(i) + '-' + str(j) + '-' + str(k) if i != 1 or j == 2 else 'ccc' for i in range(0,2) for j in range(0,3) for k in range(0,4)]

print(C)
#['0-0-0', '0-0-1', '0-0-2', '0-0-3', '0-1-0', '0-1-1', '0-1-2', '0-1-3', '0-2-0', '0-2-1', '0-2-2', '0-2-3', 'ccc', 'ccc', 'ccc', 'ccc', 'ccc', 'ccc', 'ccc', 'ccc', '1-2-0', '1-2-1', '1-2-2', '1-2-3'] 

参考記事によれば、内包表記は多重ループでもできるらしいので、エラーが出た原因は多重ループではなく単純なコードミスだと思います。

参考記事：【Python】リスト内包表記 x 多重ループ

ご参考になれば幸いです。

Comments

[mega_ne]

単純なコードミスでした...
ありがとうございます！