Python-正确删除列表中的指定列表

title: Python-删除列表中的列表
date: 2020-09-14 22:12:14
tags: Python采坑记

假设这样一个场景，有一个列表 total_list, 里面包含的三个元素都是列表，分别是 l1, l2 和 l3, 假设现在给定其中的一个元素，比如 l1, 要从 total_list 中删除它。第一时间想到的会是使用下面的语句 total_list.remove(l1)，乍一看很合情合理。并且通过下面的测试语句也可正常运行

In [1]: l1 = [1, 2]

In [2]: l2 = [3, 4]

In [3]: l3 = [5, 6]

In [4]: total_list = [l1, l2,  l3]

In [5]: print(total_list)
[[1, 2], [3, 4], [5, 6]]

In [6]: total_list.remove(l1)

In [7]: print(total_list) # 正常删除 l1
[[3, 4], [5, 6]]

In [8]: total_list.remove(l2)

In [9]: print(total_list) # 正常删除 l2
[[5, 6]]

In [10]: total_list.remove(l3)

In [11]: print(total_list) # 正常删除 l3
[]

不知各位有没有看出什么异常，如果看不出的话，不妨看下下面的另一个测试例子

In [1]: l1 = []

In [2]: l2 = []

In [3]: l3 = []

In [4]: total_list = [l1, l2, l3]

In [5]: id(l1), id(l2), id(l3)
Out[5]: (140110036119496, 140110036107848, 140110036127368)

In [6]: total_list.remove(l3)

In [7]: print(total_list) # 可以看出 total_list 的列表确实少了一个
[[], []]

In [8]: l3.append(1) # 按照设想，l3 此时已不在 total_list 中，往 l3 中添加内容，应该不影响 total_list

In [9]: print(total_list) # 可以看出 total_list 改变了, l3 还在 total_list 中
[[], [1]]


In [11]: id(total_list[0]), id(total_list[1]) # 查看 id, 发现 l2 和 l3 还在 total_list 中，被删除的其实是 l1 !
Out[11]: (140110036107848, 140110036127368)

可以看出,我们希望删除的是 l3, 但是实际上删除的却是 l1

原因在于在遍历 total_list 时, 遍历到的第一个元素是 l1 时, 满足 __eq__ 方法, 所以此时直接把 l1 从 total_list 中删除。

Python 官方文档的解释如下

list.remove(x)
Remove the first item from the list whose value is equal to x. It raises a ValueError if there is no such item.

那么，假设我现在就想删除列表l3，可以使用如下函数

def remove_list(total_list, search_list):
     for i, _list in enumerate(total_list):
         if _list is search_list:
             del total_list[i]
             break

测试例子如下

In [1]: l1, l2, l3 = [], [], []

In [2]: id(l1), id(l2), id(l3)
Out[2]: (140163609930568, 140163610222216, 140163609929224)

In [3]: total_list = [l1, l2, l3]

In [4]: def remove_list(total_list, search_list):
   ...:     for i, _list in enumerate(total_list):
   ...:         if _list is search_list:
   ...:             del total_list[i]
   ...:             break
   ...:

In [5]: remove_list(total_list, l3)

In [6]: total_list
Out[6]: [[], []]

In [7]: l3.append(1)

In [8]: total_list
Out[8]: [[], []]

In [9]: id(total_list[0]), id(total_list[1])
Out[9]: (140163609930568, 140163610222216)

In [10]: 

可以看出 l3 已经从 total_list 中删除了