多个列表根据交集进行合并
# -*- coding: utf-8 -*-#
# -------------------------------------------------------------------------------
# Name: 列表根据交集合并
# Author: yunhgu
# Date: 2022/2/22 15:16
# Description:
# -------------------------------------------------------------------------------
def merge_lists1(list_2d: list):
"""
:param list_2d:二维列表
:return:
"""
sets = {}
for lst in list_2d:
s = set(lst)
t = set()
for x in s:
if x in sets:
t.update(sets.get(x))
else:
sets[x] = s
for y in t:
sets[y] = s
s.update(t)
result_list = []
for s in sets.values():
if list(s) not in result_list:
result_list.append(list(s))
return result_list
def merge_lists2(lists_2d: list):
set_list = list(map(set, lists_2d))
i, n = 0, len(set_list)
while i < n - 1:
for j in range(i + 1, n):
if set_list[i].intersection(set_list[j]):
set_list[i].update(set_list[j])
del set_list[j]
n -= 1
break
else:
i += 1
return [sorted(i) for i in set_list]
if __name__ == '__main__':
data_list = [[1, 2], [4, 5], [1, 6], [2, 9]]
print(merge_lists1(data_list))
print(merge_lists2(data_list))
不论你在什么时候开始,重要的是开始之后就不要停止。
不论你在什么时候结束,重要的是结束之后就不要悔恨。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 单元测试从入门到精通
· 上周热点回顾(3.3-3.9)
· winform 绘制太阳,地球,月球 运作规律