如何使用任意匹配函数对Python列表进行差异/对齐?

时间:2010-04-27 04:49:02

标签: python diff

我想以类似于difflib.Differ的方式对齐两个列表,除了我希望能够定义用于比较项的匹配函数,而不仅仅是使用字符串相等,最好是匹配函数可以返回介于0.0和1.0之间的数字,而不仅仅是布尔值。

所以,例如,假设我有两个列表:

L1 = [('A', 1), ('B', 3), ('C', 7)]
L2 = ['A', 'b', 'C']

我希望能够写出这样的匹配函数:

def match(item1, item2):
    if item1[0] == item2:
        return 1.0
    elif item1[0].lower() == item2.lower():
        return 0.5
    else:
        return 0.0

然后执行:

d = Differ(match_func=match)
d.compare(L1, L2)

并使用匹配函数进行区分。就像difflib一样,我宁愿算法给出更直观的Ratcliff-Obershelp类型结果,而不是纯粹的最小Levenshtein距离。

2 个答案:

答案 0 :(得分:8)

我刚刚编写了Needleman-Wunsch的这个实现,它似乎做了我想要的事情:

def nw_align(a, b, replace_func, insert, delete):

    ZERO, LEFT, UP, DIAGONAL = 0, 1, 2, 3

    len_a = len(a)
    len_b = len(b)

    matrix = [[(0, ZERO) for x in range(len_b + 1)] for y in range(len_a + 1)]

    for i in range(len_a + 1):
        matrix[i][0] = (insert * i, UP)

    for j in range(len_b + 1):
        matrix[0][j] = (delete * j, LEFT)

    for i in range(1, len_a + 1):
        for j in range(1, len_b + 1):
            replace = replace_func(a[i - 1], b[j - 1])
            matrix[i][j] = max([
                (matrix[i - 1][j - 1][0] + replace, DIAGONAL),
                (matrix[i][j - 1][0] + insert, LEFT),
                (matrix[i - 1][j][0] + delete, UP)
            ])

    i, j = len_a, len_b
    align_a = ""
    align_b = ""

    while (i, j) != (0, 0):
        if matrix[i][j][1] == DIAGONAL:
            align_a += a[i - 1]
            align_b += b[j - 1]
            i -= 1
            j -= 1
        elif matrix[i][j][1] == LEFT:
            align_a += "-"
            align_b += b[j - 1]
            j -= 1
        else: # UP
            align_a += a[i - 1]
            align_b += "-"
            i -= 1

    return align_a[::-1], align_b[::-1]

答案 1 :(得分:0)

我最近遇到了一个名为patience diff的算法的讨论,听起来很简单。您可以尝试自己实现,然后当然可以使用您喜欢的任何比较算法。