作者mikapauli (cablin)
看板Python
标题Re: [问题] 动态产生FOR回圈的办法
时间Sun Dec 18 03:03:37 2011
我的写法：
def product(lists):
indices = range(len(lists))
iters = [iter(l) for l in lists]
values = [next(i) for i in iters]
while True:
yield values
for i in indices:
try:
values[i] = next(iters[i])
break
except StopIteration:
iters[i] = iter(lists[i])
values[i] = next(iters[i])
else:
break
>>> L
(['a', 'b', 'c'], ['d', 'e', 'f'], ['g', 'h', 'i'])
>>> [''.join(s) for s in product(L)]
['adg', 'bdg', 'cdg', 'aeg', 'beg', 'ceg', 'afg', 'bfg', 'cfg', 'adh', 'bdh',
'cdh', 'aeh', 'beh', 'ceh', 'afh', 'bfh', 'cfh', 'adi', 'bdi', 'cdi', 'aei',
'bei', 'cei', 'afi', 'bfi', 'cfi']
程式新手，请多指教<(_ _)>
--
「死ぬのっていや？ 杀されたくない？」
--
※ 发信站: 批踢踢实业坊(ptt.cc)
◆ From: 220.132.236.23