Понимание списка Python для возвращения граничных значений списка
Если у меня есть список в Python, таком как:
stuff = [1, 2, 3, 4, 5, 6, 7, 8, 9]
с длиной n (в этом случае 9) и я интересуюсь созданием списков длины n/2 (в этом случае 4). Я хочу все возможные множества значений n/2 в исходном списке, например:
[1, 2, 3, 4], [2, 3, 4, 5], ..., [9, 1, 2, 3]
есть ли некоторый код понимания списка, который я мог использовать, чтобы выполнить итерации через список и получить все те подсписки? Я не забочусь о порядке значений в рамках списков, я просто пытаюсь найти умный метод генерации списков.
3 ответа
>>> stuff = [1, 2, 3, 4, 5, 6, 7, 8, 9]
>>>
>>> n=len(stuff)
>>>
>>> [(stuff+stuff[:n/2-1])[i:i+n/2] for i in range(n)]
[[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6], [4, 5, 6, 7], [5, 6, 7, 8], [6, 7, 8, 9], [7, 8, 9, 1], [8, 9, 1, 2], [9, 1, 2, 3]]
>>>
Примечание : приведенный выше код основан на предположении из вашего примера
[1, 2, 3, 4], [2, 3, 4, 5], ..., [9, 1, 2, 3]
Если вам действительно нужны все возможные значения, вам нужно использовать itertools.permutations или комбинацию функций, как предлагали другие.
Вам нужна функция комбинаций из itertools (РЕДАКТИРОВАТЬ: используйте перестановку, если порядок важен)
Обратите внимание, что эта функция недоступна в Python 2.5. В этом случае вы можете скопировать код из приведенной выше ссылки:
def combinations(iterable, r):
# combinations('ABCD', 2) --> AB AC AD BC BD CD
# combinations(range(4), 3) --> 012 013 023 123
pool = tuple(iterable)
n = len(pool)
if r > n:
return
indices = range(r)
yield tuple(pool[i] for i in indices)
while True:
for i in reversed(range(r)):
if indices[i] != i + n - r:
break
else:
return
indices[i] += 1
for j in range(i+1, r):
indices[j] = indices[j-1] + 1
yield tuple(pool[i] for i in indices)
, а затем
stuff = range(9)
what_i_want = [i for i in combinations(stuff, len(stuff)/2)]
Используйте itertools.permutations () или itertools.combinations () (в зависимости от того, хотите ли вы, например, оба [1,2,3,4] и [4,3,2,1] или нет) с необязательным вторым аргументом для указания длины.
stuff = [1, 2, 3, 4, 5, 6, 7, 8, 9]
itertools.permutations(stuff, 4) # will return all possible lists of length 4
itertools.combinations(stuff, 4) # will return all possible choices of 4 elements
Предполагается, что вам нужны не только смежные элементы.
Обновление
Поскольку вы указали, что вас не волнует порядок, вы, вероятно, ищете itertools.combinations () .