内部的Python枚举for循环跳过备用项

时间:2013-11-23 11:50:38

标签: python for-loop enumerate

我有一种情况,类外的for循环似乎与类中的枚举进行交互。

示例代码:

class MyContainerClass(object):
    def __init__(self):
        self.collection = []
    def __str__(self):
        return (len(self.collection) == 0 and 'Empty' or
                ', '.join(str(item) for item in self.collection))
    def __iadd__(self,item):
        print "adding {item}".format(item=str(item))
        self.collection.append(item)
        return self
    def __iter__(self):
        return iter(self.collection)
    def discard(self,item):
        print "discarding {item}".format(item=str(item))
        for index, value in enumerate(self.collection):
            if value == item:
                print "found {value}=={item}".format(value=value,item=item)
                return self.collection.pop(index)
        return False

class MyItemClass(object):
    def __init__(self,value):
        self.value=value
    def __str__(self):
        return '{value}'.format(value=self.value)
    def __eq__(self,other):
        if self.value == other.value:
            return True
        else:
            return False

c1 = MyContainerClass()
c2 = MyContainerClass()

c2 += MyItemClass('item1')
c2 += MyItemClass('item2')
c2 += MyItemClass('item3')
c2 += MyItemClass('item4')
c2 += MyItemClass('item5')
print "c1 is : {c1}".format(c1=str(c1))
print "c2 is : {c2}".format(c2=str(c2))

for item in c2:
    print "for got {item}".format(item=str(item))
    c1 += c2.discard(item)

print "c1 is : {c1}".format(c1=str(c1))
print "c2 is : {c2}".format(c2=str(c2))

生成此输出:

adding item1
adding item2
adding item3
adding item4
adding item5
c1 is : Empty
c2 is : item1, item2, item3, item4, item5
for got item1
discarding item1
found item1==item1
adding item1
for got item3
discarding item3
found item3==item3
adding item3
for got item5
discarding item5
found item5==item5
adding item5
c1 is : item1, item3, item5
c2 is : item2, item4

我确定这是非常明显的,可能与 iter 功能有关,但我现在看不到它。

2 个答案:

答案 0 :(得分:0)

您正在修改c2.collection(在discard中),同时对其进行迭代(for item in c2:)。别这么做。

答案 1 :(得分:0)

在迭代容器时修改容器是不安全的。

在这种情况下,正在发生的事情是迭代只是跟踪已经过去的元素数量。因此当你取出item1时,迭代器知道它已经超过了容器中的第一个项目......现在是item2。因此,它很高兴继续item3