我想以类似于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距离。
答案 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的算法的讨论,听起来很简单。您可以尝试自己实现,然后当然可以使用您喜欢的任何比较算法。