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 中删除了
