@[toc]
需求及背景
我有一个需求:保留list_1中含有list_2元素的元素,其他的删除
list_1 = ['魁哥','夭夭', '阿离', '渣姐', '秃头企鹅', '王哥', '小白', '卷魔']
list_2 = ['旺仔', '秃头企鹅', '王哥']
最后需要输出
['秃头企鹅', '王哥']
直接一顿操作,写出了如下代码
for i in list_1:
if i not in list_2:
list_1.remove(i)
print(list_1)
结果输出的结果是
['夭夭', '渣姐', '秃头企鹅', '王哥', '卷魔']
为什么会这样?通过debug发现,每删除一个元素,list_1里面的元素下标就会发生变化,而for循环里面是持续性的读取下一个值的,每次删除一个的时候,便会跳过一个。
既然如此我就是用下标删呗
for i in range(len(list_1)):
if list_1[i] not in list_2:
list_1.remove(list_1[i])
continue
print(list_1)
这时候又发现,直接报错
IndexError: list index out of range
当删除一个元素之后,新列表的长度减少,i 还是根据原来列表的索引取值,当 i 超过当前的列表长度时,新列表没有该下标对应元素,索引越界。
解决办法
方案一(快捷)
list_1 = ['夭夭', '阿离', '渣姐', '秃头企鹅', '魁哥', '小白', '卷魔']
list_2 = ['旺仔', '秃头企鹅', '王哥']
j = 0
for i in range(len(list_1)):
if list_1[j] not in list_2:
list_1.remove(list_1[j])
continue
j+=1
print(list_1)
方案一方法的思路:
每次只要删除了,都从上一次的下标继续,如果这次没删除,则下标加一删下一个。看不太懂的可以丢自己编辑器里面加一些print进行分步分析。
输出
['秃头企鹅', '王哥']
方案二(简单)
*倒序
如果让索引倒序遍历列表就不会出现越界的问题了。*
for i in range(len(list_1)-1,-1,-1):
if list_1[i] not in list_2:
list_1.remove(list_1[i])
print(list_1)
方案二方法思路:
将索引倒序,这样不管你怎么删,我下标都是倒着来,肯定不会下标越界。(此方法对应解决最开始的下标越界的思路。)
注意细节:(自己多尝试,多写)
①len减一是因为左闭右开;
②第三个参数表示的是对数值所进行的操作,每次加上-1;
③第二个值表示最终加到-1,但不包含-1。(最终加到几,但是不包含本身)
输出
['秃头企鹅', '王哥']
方案三(粗暴)
while True:
for i in list_1:
if i not in list_2:
list_1.remove(i)
break
else:
break
print(list_1)
方案三思路:
无限循环,知道你搞完,这个不多解释,就是莽夫。
输出
['秃头企鹅', '王哥']
方案四(灵活)
d = dict(zip(range(len(list_1)),list_1))
for k,v in d.items():
if v not in list_2:
d[k]=[]
final_1 = [v for k,v in d.items() if v]
print(final_1)
方案四思路:
讲列表转换成字典,再通过字典的value和list_2进行对比,如果不是list_2中的元素则赋值为空列表,最后拿到不是空列表的value,就是我们需要的列表了。
听起来会有点绕,自己尝试一下,一下就练了字典、列表、遍历、推导式、列表合并字典这些知识点哦。
输出
['秃头企鹅', '王哥']
上面用例列表推导式,zip,dict等方法,这些python基础我之前文章有写,大家可以多翻翻。(如果没有,那肯定是我不小心删了。)
最后,祝大家周末愉快!
gzh:梦无矶的测试开发之路