Изменение набора во время итерации по нему

Question

Я хотел бы перебрать набор, пока я удаляю элементов из него. Есть похожие вопросы для deleting one item at the time или для lists, но они не работают для моего дела.

Код следует; Я перебираю набор ZN, и в конце итерации я удаляю несколько предметов (принадлежащих набору temp). Но итерация все еще происходит над «оригинальным» ZN.

Как изменить этот код, чтобы изменить набор ZN в то время как Я перебираю его?

def CyclotomicCosets(q,n): 
    N=q^n-1 
    ZN=set(range(N)) 
    Cosets=[] 
    for i in ZN: 
     tmp=set([]) 
     for j in range(n): 
      tmp.add(i*(q^j) %N) 
     Cosets.append(list(tmp)) 
     ZN=ZN.difference(tmp) # <------------ Does not do what I want 
    return(Cosets) 

Answer 1

Используйте while петли и .pop() значение для обработки из набора:

def CyclotomicCosets(q, n): 
    N = q^n - 1 
    ZN = set(range(N)) 
    Cosets = [] 
    while ZN: 
     i = ZN.pop() 
     tmp = {i * (q^j) % N for j in range(n)} 
     Cosets.append(list(tmp)) 
     ZN -= tmp 
    return Cosets 

Обратите внимание, что я заменил свой внутренний for цикл с заданным пониманием, чтобы сделать его немного быстрее и компактнее. Они были введены в Python 2.7 и Python 3, в более ранних версиях Python вы можете использовать выражение генератора вместо:

tmp = set(i * (q^j) % N for j in range(n)) 

Ваша первоначальная ошибка была заменитьZN, а не обновлять его:

ZN=ZN.difference(tmp) 

Это не изменило исходный набор, который вы использовали в цикле for. Скорее, вы создаете новый набор и указываете на него ссылку ZN.

Однако вы не можете изменить set во время итерации по нему, поэтому даже разница на месте не сработала бы; вы должны использовать ZN -= tmp или ZN.difference_update(tmp), но это приведет к исключениям вместо:

>>> ZN = set(range(3)) 
>>> for i in ZN: 
...  ZN -= set([2]) 
... 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
RuntimeError: Set changed size during iteration 

Исправленный код дает:

>>> CyclotomicCosets(3, 5) 
[[0], [0, 1, 2, 3], [0, 1, 4, 5], [0, 4, 5, 6]] 

Альтернативно, цикл по range(N) вместо этого, и сохранить набор значений вас» ве уже обработано:

def CyclotomicCosets(q, n): 
    N = q^n - 1 
    Cosets = [] 
    seen = set() 
    for i in range(N): 
     if i in seen: continue 
     tmp = {i * (q^j) % N for j in range(n)} 
     Cosets.append(list(tmp)) 
     seen |= tmp 
    return Cosets