将列表中有相同元素的集合合并（保留值相同的重复元素，不想保留+sorted）

最新推荐文章于 2024-04-30 15:36:09 发布

YnagShanwen

最新推荐文章于 2024-04-30 15:36:09 发布

阅读量1.5k

点赞数 2

文章标签： python 列表数据结构合并排序

本文链接：https://blog.csdn.net/YnagShanwen/article/details/111344386

版权

python 多层列表合并有相同数值的元素

问题描述：假如有一个列表，列表中有许多集合，集合之间可能有重复的元素，设计一个算法，将有相同元素的集合进行合并,并且保留值相同的重复元素。
问题实际应用场景：成对信息在分组后需要保留的情况。
如不需保留，将分组后用sorted函数合并相同元素

#定义分组函数
def grouping(bondlist):
    groups = [] 
    break1 = False    
    while bondlist:
        pair1 = bondlist.pop(0)        
        a = 1111 ##判断是否循环的初始值，设成这么这么大是为了不和列表中元素个数碰巧相等
        b = 1000
        while b != a:
            a = b
            for atomid in pair1:
                for i,pair2 in enumerate(bondlist):            
                    if atomid in pair2:
                        pair1 = pair1 + pair2
                        bondlist.pop(i)
                        if not bondlist:
                            break1 = True
                        break
                if break1:
                    break
            b = len(pair1)
            print(pair1) #测试分类过程和结果用
            print(b) #测试分类过程和结果用
        groups.append(pair1)    
    return groups

import copy
BKN_BOND_LIST = [[1,2],[3,5],[2,3],[7,5],[10,20]]
BKN_BONDS = copy.deepcopy(BKN_BOND_LIST)
grouping(BKN_BONDS)

运行结束后，此时的BKN_BONDS = [] 为空集（所有元素都pop掉了），若后续仍能用到初始的BKN_BOND_LIST列表中的信息，则在操作前对多层list进行deepcopy。

运行过程：
[1, 2, 2, 3]
4
[1, 2, 2, 3, 3, 5]
6
[1, 2, 2, 3, 3, 5, 7, 5]
8
[1, 2, 2, 3, 3, 5, 7, 5]
8
[10, 20]
2
[10, 20]
2
运行结果：
out: [[1, 2, 2, 3, 3, 5, 7, 5], [10, 20]]