Быстрые/более умные петли [Python]

Question

Файл данных - это список Excel с 4 столбцами: имя, должность, значение и цена. Файл, импортированный с помощью pandas.

Существует 5 различных положений.

Я стараюсь соответствовать всем различным возможностям, где он может быть только одним из каждой позиции + одним случайным игроком, который еще не выбран.

def teams(x): 
    positions = list(data.Position.unique()) 
    indices = [] 
    for position in positions: 
     indices.append((data.Position == position).nonzero()[0]) 

    positions.append('Random') 
    indices.append(data.index) 


    for i in indices[0]: 
     for i2 in indices[1]: 
      for i3 in indices[2]: 
       for i4 in indices[3]: 
        for i5 in indices[4]: 
         for i6 in indices[5]: 
          if i6 != i: 
           if i6 != i2: 
            if i6 != i3: 
             if i6 != i4: 
              if i6 != i5: 
               print i, i2, i3, i4, i5, i6 

teams(data) 

Моя проблема заключается в том, что мои петли слишком медленно, и я не могу получить if заявление на работу, так что я должен сделать 5 различных if с, что я предполагаю, что это довольно глупо решение.

Мой вопрос: как я могу сделать мои петли быстрее/умнее и как это исправить мое if заявление

Answer 1

Вы можете сократить свой код на три строки, если вы используете itertools.product() как так

from itertools import product 

for i in list(product(*indices)): 
    if i[5] not in i[:5]: 
     print i[0], i[1], i[2], i[3], i[4], i[5] 

product() эквивалентен вложенным для циклов