迭代修改集合

时间:2013-05-18 15:33:59

标签: python set iteration sage

我想在删除时迭代一组 来自它的物品。 deleting one item at the timelists也存在类似问题,但确实如此 不适合我的情况。

代码如下;我迭代了一个 设置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)

1 个答案:

答案 0 :(得分: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 -= tmpZN.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