В python есть более простой способ написать 6 вложенных циклов for?

эта проблема добирается до меня в течение некоторого времени. Есть ли более простой способ написания вложенных for петли в python? Например, если мой код что-то вроде этого:

  for y in range(3):
    for x in range(3):
      do_something()
      for y1 in range(3):
        for x1 in range(3):
          do_something_else()

будет ли более простой способ сделать это? Я знаю, что этот код работает, но при понижении уровня вместо 2 места, как я, это может стать проблемой.

О в Примере было только 4 вложенных for петли, чтобы сделать вещи проще.

11 ответов


Если вы часто повторяете декартово произведение, как в вашем примере, вы можете исследовать itertools Python 2.6.продукт -- или напишите свой собственный, если вы находитесь в более раннем Python.

from itertools import product
for y, x in product(range(3), repeat=2):
  do_something()
  for y1, x1 in product(range(3), repeat=2):
    do_something_else()

Это довольно распространено при циклическом выполнении многомерных пространств. Мое решение:

xy_grid = [(x, y) for x in range(3) for y in range(3)]

for x, y in xy_grid:
    # do something
    for x1, y1 in xy_grid:
        # do something else

столкнувшись с такого рода программной логикой, я, вероятно, разбил бы последовательность циклов на две или более отдельных функций.

другой метод в Python-использовать списочные включения там, где это возможно, вместо петли.


предполагая, что каждый цикл имеет какое-то независимое значение, разбейте их на именованные функции:

def do_tigers():
    for x in range(3):
        print something

def do_lions():
    do_lionesses()
    for x in range(3):
        do_tigers()

def do_penguins():
    for x in range(3):
        do_lions()

..etc.

возможно, я мог бы выбрать лучшие имена. 8-)


технически, вы можете использовать itertools.product чтобы получить декартово произведение N последовательностей и повторить это:

 for y, x, y1, x1 in itertools.product(range(3), repeat=4):
   do_something_else()

но я не думаю, что на самом деле выигрывает у вас что-нибудь читабельность.


итераторы Python и генераторы, в частности, существуют именно для того, чтобы обеспечить хороший рефакторинг в противном случае сложных циклов. Конечно, трудно получить абстракцию из простого примера, но предполагая 3 должен быть параметром (возможно, целым range(3) должно быть?), и две функции, которые вы вызываете, нуждаются в некоторых параметрах, которые являются переменными цикла, вы можете рефакторировать код:

  for y in range(3):
    for x in range(3):
      do_something(x, y)
      for y1 in range(3):
        for x1 in range(3):
          do_something_else(x, y, x1, y1)

, например:

def nestloop(n, *funcs):
  head = funcs[0]
  tail = funcs[1:]
  for y in range(n):
    for x in range(n):
      yield head, x, y
      if tail:
        for subtup in nestloop(n, *tail):
           yield subtup[:1] + (x, y) + subtup[1:]

for funcandargs in nestloop(3, do_something, do_something_else):
  funcandargs[0](*funcandargs[1:])

точный вид рефакторинг, несомненно, нужно будет настроить для ваших точных целей, но общий момент, что итераторы (и, как правило, просто простые генераторы) позволяют очень хорошие рефакторинги циклов остается-вся логика цикла идет внутри генератора, и код уровня приложения остается с простым for петли и фактическое применение-соответствующая обработка элементов, полученных в циклах for.


моим личным аргументом было бы то, что вы, вероятно, делаете что-то неправильно, если у вас есть 6 вложенных циклов...

тем не менее, функциональная декомпозиция-это то, что вы ищете. Рефакторинг, поэтому некоторые из циклов происходят в отдельных вызовах функций, а затем вызывают эти функции.


из вашего кода похоже, что вы хотите выполнить операцию с каждой возможной парой точек, где x и y находятся в диапазоне 0..2.

для этого:

for x1,y1,x2,y2 in itertools.product(range(3), repeat=4):
    do_something_with_two_points(x1,y1,2,y2)

операция do_something_with_two_points будет вызываться 81 раз-один раз для каждой возможной комбинации точек.


вы заглянули в Списочные Включения?

что-то типа:

[do_something() for x in range(3) for y in range(3)]

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

другой вариант, о котором я мог бы подумать, - использовать функцию для генерации параметров, а затем просто применить их в цикле

def generate_params(n):
    return itertools.product(range(n), range(n))

for x,y in generate_params(3):
    do_something()

вы также можете использовать map () функции