python - 子列表具有共同元素的列表的合并列表

Question

我有一个这样的列表列表

list = [[1, 2], [1, 3], [4, 5]]

如您所见，前两个子列表的第一个元素被重复

所以我希望我的输出也是:

list = [[1, 2, 3], [4, 5]]

谢谢

Answer 1

下面的代码应该可以解决您的问题:

def merge_subs(lst_of_lsts):
    res = []
    for row in lst_of_lsts:
        for i, resrow in enumerate(res):
            if row[0]==resrow[0]:
                res[i] += row[1:]
                break
        else:
            res.append(row)
    return res

请注意，else 属于内部 for，如果循环没有中断就被执行。