Перемещение элементов 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]