python - 从列表中删除某些项目重复的列表

Question

我试图从列表中删除列表，该列表具有相同的第一个和第三个项目但只保留第一个。示例列表和输出:

li=[ [2,4,5], [1,3,5], [1,6,5] ]
output_list = [ [2,4,5], [1,3,5] ]

我编写的代码需要很长时间才能执行，因为原始列表包含数百万个列表。

b_li = []
output_list = []
for x in li:
    s = [ x[0], x[2] ]
    if s not in b_li:
        b_li.append(s)
        output_list.append(x)

如何改进代码？提前致谢。

Answer 1

改进版:

b_li = set()
output_list = []
b_li_add = b_li.add
output_list_append = output_list.append
for x in li:
    s = (x[0], x[2])
    if s not in b_li:
        b_li_add(s)
        output_list_append(x)

变化是:

• 对 b_li 使用 set() 可以加快查找速度。
• 将 s 转换为元组，因为不需要将唯一的第一个和第三个元素存储为列表。
• 减少了函数查找，这也加快了代码速度。