Алгоритм подстановки подстановки в Python и печатных строках

Question

Я уже делал этот алгоритм в школе, но я забыл, как это сделать. В основном я хочу, чтобы вернуть результат, которые являются строки, как 'в [0]', 'в [0] .a [0]' ...

length = range(0,2) #length = 2 
depth = range(0,3) #depth = 3 

for i in length: 
    for k in depth: 
     … print each permutation 

РЕЗУЛЬТАТ

a[0] 
a[0].a[0] 
a[0].a[1] 
a[0].a[0].a[0] 
a[0].a[0].a[1] 
a[0].a[1].a[0] 
a[0].a[1].a[1] 
a[1] 
a[1].a[0] 
a[1].a[1] 
a[1].a[0].a[0] 
a[1].a[0].a[1] 
a[1].a[1].a[0] 
a[1].a[1].a[1] 

Answer 1

Изменение немного упорядочение производства, так что это одно и то же на всех уровнях:

def thing (length, depth, prefix = None): 
    if not depth: return 
    if not prefix: prefix = [] 
    for l in range (length): 
     r = prefix + ['a[{}]'.format (l) ] 
     yield '.'.join (r) 
     for r in thing (length, depth - 1, r): 
      yield r 

for x in thing (2, 3): print (x) 

---

выход:

a[0] 
a[0].a[0] 
a[0].a[0].a[0] 
a[0].a[0].a[1] 
a[0].a[1] 
a[0].a[1].a[0] 
a[0].a[1].a[1] 
a[1] 
a[1].a[0] 
a[1].a[0].a[0] 
a[1].a[0].a[1] 
a[1].a[1] 
a[1].a[1].a[0] 
a[1].a[1].a[1] 

Answer 2

Альтернативный подход; как указывает Гипербореус, ключ понимает, что упорядочение первых элементов не соответствует порядку следующих элементов; поэтому я обрабатываю их отдельно.

from itertools import product 

def item_fmt(i): 
    return "a[{}]".format(repr(i)) 

def make_result(*args): 
    return ".".join(item_fmt(arg) for arg in args) 

def main(): 
    items = [0, 1] 
    maxdepth = 3 

    for first in items:    # in order by first-item 
     print(make_result(first)) # show first-item-only 
     for depth in range(1, maxdepth):    # in order by depth 
      for combo in product(items, repeat=depth): # generate all combinations of given depth 
       print(make_result(first, *combo)) 

if __name__=="__main__": 
    main() 

приводит

a[0] 
a[0].a[0] 
a[0].a[1] 
a[0].a[0].a[0] 
a[0].a[0].a[1] 
a[0].a[1].a[0] 
a[0].a[1].a[1] 
a[1] 
a[1].a[0] 
a[1].a[1] 
a[1].a[0].a[0] 
a[1].a[0].a[1] 
a[1].a[1].a[0] 
a[1].a[1].a[1]