Python 离线等价类

离线等价类的概念见离线等价类

最近在清洗数据的时候涉及到要将相似度比较高的文件夹合并，特征比对得到是1:1的对，比如：

（a,b),(c,d),(a,c)...,那么合并的时候就涉及到将这些等价的对合并成一个大类，直观上这就是一个离线等价类的问题。

import os
import pickle
with open(r"D:\PhotoLibrary\Uface_cropped_pairs\same_pairs0.8.pkl",'rb') as f:
    pairs = pickle.load(f)
#construct pair list
pairs_lst={}
for pair in pairs:
    a,b = pair
    if a not in pairs_lst:
        pairs_lst[a]=[]
    if b not in pairs_lst:
        pairs_lst[b]=[]
    pairs_lst[a].insert(0,b)
    pairs_lst[b].insert(0,a)
flag={}
classes=[]
S=[]
for item in pairs_lst.items():
    if not item[0] in flag:
        flag[item[0]]=True
        set=[item[0]]
        S.append(item[0])
        while len(S)>0:
            a=S.pop()
            for b in pairs_lst[a]:
                if not b in flag:
                    flag[b]=True
                    S.append(b)
                    set.append(b)
        classes.append(set)
for s in classes:
    print(s)
print("test")