在列表之間移動Python元素

Question

listA = [1,2,3] 
listB = [] 

print listA 
print listB 

for i in listA: 
     if i >= 2: 
       listB.append(i) 
       listA.remove(i) 
print listA 
print listB 

爲什麼只能添加和移除元素「2」？

另外，當我註釋掉「listA.remove（i）」時，它按預期工作。

Answer 1

您不應該修改您正在迭代的列表，這會導致令人驚訝的行爲（因爲迭代器在內部使用索引，並且通過刪除元素來更改這些索引）。你可以做的是要遍歷一個副本listA的：

for i in listA[:]: 
    if i >= 2: 
    listB.append(i) 
    listA.remove(i) 

例子：

>>> listA = [1,2,3] 
>>> listB = [] 
>>> for i in listA[:]: 
... if i >= 2: 
...  listB.append(i) 
...  listA.remove(i) 
... 
>>> listA 
[1] 
>>> listB 
[2, 3] 

然而，它往往是乾淨多了去了功能性的方式，而不是修改原來的列表中的所有，而是隻需創建一個包含所需值的新列表。您可以使用列表理解這樣做優雅：

>>> lst = [1,2,3] 
>>> small = [a for a in lst if a < 2] 
>>> big = [a for a in lst if a >= 2] 
>>> small 
[1] 
>>> big 
[2, 3]