Как перебирать списки из середины

Question

Я хотел бы перебрать все списки/кортежи длины n с элементами из -s ... s. В настоящее время я делаю это с:

for k in itertools.product(range(-s,s+1), repeat = n): 
    #process k and maybe print out the result 

Однако это не полезно для меня, поскольку есть огромное количество таких кортежей и мой код никогда не может завершиться. Я бы очень хотел начать сначала с самых интересных. В этом случае порядок я хотел бы для итерации:

1. Все кортежи, которые содержат только 0 (есть только один)
2. Все кортежи, которые содержат только 0, 1 и -1 за исключением тех кортежей мы имеем уже видел.
3. Все кортежи, содержащие только 0, 1, -1, 2 и-2, исключая те кортежи, которые мы уже видели.
4. И так далее ...

Как можно это сделать?

Answer 1

Значит, ваш код будет со списком намного дороже сортировки?

Затем вы можете отсортировать список из этих списков с аргументом key. То, что вы называете списком, - это кортежи, верно? По крайней мере, в моем python 2.7 itertools. Я бы преобразовал их в массивы, так как я думаю, что вы не можете использовать abs в противном случае. Тогда функция сортировки:

lists.sort(key = lambda t: np.max(np.abs(np.array(t)))) 

Это работает достаточно быстро?

Answer 2

Как об этом:

import itertools 

def sorted_tuples(length, max_s): 
    nums = [0] 
    for s in range(max_s): 
     for p in itertools.combinations_with_replacement(nums, length): 
      if s in p or -s in p: 
       yield p 
     nums = [-(s+1)] + nums + [s+1] 

for i in sorted_tuples(3,2): 
    print(i) 

# prints the following 
(0, 0, 0) 
(-1, -1, -1) 
(-1, -1, 0) 
(-1, -1, 1) 
(-1, 0, 0) 
(-1, 0, 1) 
(-1, 1, 1) 
(0, 0, 1) 
(0, 1, 1) 
(1, 1, 1)