通过difflib modul比较python中的列表

时间:2011-03-30 16:43:35

标签: python difflib

我正在尝试使用difflib库。我有两个列表:L_1和L_2包含字符串。我想知道,如果这些序列相似(顺序并不重要)。

L_1 = ["Bob", "Mary", "Hans"]
L_2 = ["Bob", "Marie", "Háns"]

应该没问题。但是

L_1 = ["Nirdosch", "Mary", "Rolf"]
L_2 = ["Bob", "Marie", "Háns"]

不应该没问题。

我提出了迭代第一个列表L_1并通过方法匹配L_1的每个元素的想法

difflib.get_close_matches()

针对第二个列表L_2。如果匹配比例更大,那么让我们说0.7从L_2中删除并继续。但我怀疑这是一个好计划。还有更好的吗?

1 个答案:

答案 0 :(得分:3)

我会做类似的事情:

import difflib

L_1 = ["Bob", "Mary", "Hans"]
L_2 = ["Bob", "Marie", "Hans"]

def similiarity(L_1, L_2):
    L_1 = set(intern(w) for w in L_1)
    L_2 = set(intern(w) for w in L_2)

    to_match = L_1.difference( L_2)
    against = L_2.difference(L_1)
    for w in to_match:
        res = difflib.get_close_matches(w, against)
        if len(res):
            against.remove( res[0] )
    return (len(L_2)-len(against)) / (len(L_1))

print similiarity(L_1,L_2)