Python忽略元素顺序提取嵌套字符列表唯一子列表的最优方法
问题描述
现有一个存储字符序列的Python嵌套列表如下:
[['A', 'B'],['A','B','C'],['B','A'],['C','A','B'],['D'],['D'],['Ao','B']]
需求为忽略子列表内元素的顺序,返回所有唯一的子列表,预期输出如下:
[['A','B'],['A','B','C'],['D'],['Ao','B']]
现有实现不足
当前实现通过双层循环+元素删除判断重复,时间复杂度为O(n²*k)(n为子列表总数,k为子列表平均长度),数据量较大时执行效率很低,且写法冗余,不符合Python常用编码风格。
优化实现
针对子列表元素均为可哈希类型的场景,可以将每个子列表排序后转为元组作为唯一标识,利用集合的O(1)查找特性判重,时间复杂度降至O(n*k log k),代码更简洁:
方案1:保留子列表首次出现的顺序(推荐)
vars_list = [['A', 'B'],['A','B','C'],['B','A'],['C','A','B'],['D'],['D'],['Ao','B']] seen = set() result = [] for sub_list in vars_list: # 排序后转元组作为哈希键,可兼容子列表有重复元素的场景 unique_key = tuple(sorted(sub_list)) if unique_key not in seen: seen.add(unique_key) result.append(sub_list) print(result) # 输出:[['A', 'B'], ['A', 'B', 'C'], ['D'], ['Ao', 'B']]
方案2:Python 3.7+ 精简写法
利用3.7及以上版本字典默认保留插入顺序的特性,一行即可完成去重:
vars_list = [['A', 'B'],['A','B','C'],['B','A'],['C','A','B'],['D'],['D'],['Ao','B']] result = list({tuple(sorted(sub)): sub for sub in vars_list}.values()) print(result)
如果确认子列表不会出现重复元素,也可以用frozenset(sub_list)代替tuple(sorted(sub_list))作为唯一标识,省去排序步骤,效率更高。
内容的提问来源于stack exchange,提问作者chr218
相关产品推荐
相关产品推荐

