我有input_list = [[1,2],[1,3],[1,4],[2,8],[2,10]]
然后我有elimination_list = [[1,2],[2,8]]
然后我创建一个result_list
,其中包含输入列表的元组,但在elimination_list中找到的元组除外
因此result_list = [[1,3],[1,4],[2,10]]
我可以通过下面的多个for循环来实现这一点,但是有什么方法可以用较少的代码行来做到这一点吗?
for x,y in input_list:
is_eliminate = 0
for a,b in elimination_list:
if x == a and y == b:
is_eliminate = 1
if is_eliminate == 0:
result_list.append([x,y])
答案 0 :(得分:5)
您可以使用包含测试(in
)和列表理解:
result_list = [t for t in input_list if t not in elimination_list]
t not in elimination_list
否定收容测试会遍历所有elimination_list
,以便您查看该元素是否存在。
如果您可以将列表转换为实际具有元组,则可以使用设置成员资格测试的更快选项:
input_list = [(1, 2), (1, 3), (1, 4), (2, 8), (2, 10)]
elimination_set = {(1, 2), (2, 8)}
result_list = [t for t in input_list if t not in elimination_set]
此处eliminitaion_set
是一个包含元组(不可变序列)的集合,t not in elimination_set
使用散列算法来确定任何给定元组是否是平均O(1)(常量)时间的成员。 / p>
你不 将input_list
转换为元组,但是;只有在测试时才能进行转换:
input_list = [[1, 2], [1, 3], [1, 4], [2, 8], [2, 10]]
elimination_set = {(1, 2), (2, 8)}
result_list = [t for t in input_list if tuple(t) not in elimination_set]
答案 1 :(得分:3)
您可以将列表理解与条件一起使用:
result_list = [t for t in input_list if t not in elimination_list]
对于较大的列表,您可以使用set
来更快地消除:
elimination_set = set((tuple(x) for x in elimination_list))
result_list = [t for t in input_list if tuple(t) not in elimination_set]
功能风格的两个解决方案:
result_list = filter(lambda t: t not in elimination_list, input_list)
或:
elimination_set = set(map(lambda x: tuple(x), elimination_list))
result_list = filter(lambda t: t not in elimination_set, input_list)
答案 2 :(得分:2)
只需使用filter
和in
:
result = filter(lambda x: x not in elimination_list, input_list)