在Python中遍历列表时删除元素

时间:2009-08-30 02:18:40

标签: python list loops iterator python-datamodel

在Java中,我可以使用Iterator,然后使用迭代器的.remove()方法删除迭代器返回的最后一个元素,如下所示:

import java.util.*;

public class ConcurrentMod {
    public static void main(String[] args) {
        List<String> colors = new ArrayList<String>(Arrays.asList("red", "green", "blue", "purple"));
        for (Iterator<String> it = colors.iterator(); it.hasNext(); ) {
            String color = it.next();
            System.out.println(color);
            if (color.equals("green"))
                it.remove();
        }
        System.out.println("At the end, colors = " + colors);
    }
}

/* Outputs:
red
green
blue
purple
At the end, colors = [red, blue, purple]
*/

我将如何在Python中执行此操作?我在for循环中迭代它时无法修改列表,因为它导致跳过东西(参见here)。并且似乎没有相当于Java的Iterator接口。

4 个答案:

答案 0 :(得分:26)

Python中的最佳方法是创建一个新列表,理想情况是在listcomp中,将其设置为旧列表的[:],例如:

colors[:] = [c for c in colors if c != 'green']

不是colors =,正如一些答案可能暗示的那样 - 只会重新命名这个名字,最终会留下一些旧的“身体”悬挂的引用; colors[:] =在所有方面都要好得多; - )。

答案 1 :(得分:19)

迭代列表的副本

for c in colors[:]:
    if c == 'green':
        colors.remove(c)

答案 2 :(得分:4)

您可以使用过滤功能:

>>> colors=['red', 'green', 'blue', 'purple']
>>> filter(lambda color: color != 'green', colors)
['red', 'blue', 'purple']
>>>

答案 3 :(得分:0)

或者您也可以这样做

>>> colors = ['red', 'green', 'blue', 'purple']
>>> if colors.__contains__('green'):
...     colors.remove('green')