python 遍历list并删除部分元素

python 遍历list并删除部分元素

有两个list，list_1 为0-9，list_2 为0-4,需要删除list_1中包含在list_2中的元素

list_1 =[] for i in range(10): list_1.append(str(i)) list_1 ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'] list_2 =[] for i in range(5): list_2.append(str(i)) list_2 ['0', '1', '2', '3', '4']

为了提高执行效率，可以将大的list转成set

set_2 = set(list_2) set_2 {'0', '1', '2', '3', '4'}

错误删除方式1

直接遍历list并删除元素，这种方式会导致删除之后的元素前移，后漏掉一部分元素

temp = list_1[:] for item in temp: if item in set_2: temp.remove(item) "列表长度：%d, 列表：%s" % (len(temp), temp) "列表长度：7, 列表：['1', '3', '5', '6', '7', '8', '9']"

错误删除方式2

使用下标遍历输出，删除元素，同样也会出现删除只有的元素前移导致漏掉部分元素

temp = list_1[:] for i in range(len(temp)): try: if temp[i] in set_2: temp.pop(i) except:# 这里下标会越界，为了结果好看，不做处理 pass "列表长度：%d, 列表：%s" % (len(temp), temp) "列表长度：7, 列表：['1', '3', '5', '6', '7', '8', '9']"

正确方式1；倒序

倒序（保证 next 指向为未遍历过得）列表长度减少，但是next指向一直是为未遍历过的元素，并不会漏掉

temp = list_1[:] for i in range(len(temp)-1, -1, -1): if temp[i] in set_2: temp.pop(i) "列表长度：%d, 列表：%s" % (len(temp), temp) "列表长度：5, 列表：['5', '6', '7', '8', '9']"

正确方式2；遍历复制数组，修改原数组

这种方式能保证遍历到所有元素

temp = list_1[:] for item in temp[:]: if item in set_2: temp.remove(item) "列表长度：%d, 列表：%s" % (len(temp), temp) "列表长度：5, 列表：['5', '6', '7', '8', '9']"

正确方式3；遍历需要删除的数组

temp = list_1[:] for item in set_2: try: temp.remove(item) except: # 这里元素不存在会抛异常 pass "列表长度：%d, 列表：%s" % (len(temp), temp) "列表长度：5, 列表：['5', '6', '7', '8', '9']"

正确方式4；利用集合差集，不能保证顺序

temp = list_1[:] temp = list(set(temp).difference(set_2)) "列表长度：%d, 列表：%s" % (len(temp), temp) "列表长度：5, 列表：['8', '9', '5', '7', '6']"