7

我有一个比较两个文件的问题。基本上,我想做的是两个文件之间的类 UNIX 差异,例如:

$ diff -u 左文件右文件

但是我的两个文件包含浮点数;并且因为这些文件是在不同的架构上生成的(但计算相同的东西),所以浮动值并不完全相同(它们可能相差 1e-10)。但是我通过“差异”文件寻求的是找到我认为有显着差异的内容(例如差异超过 1e-4);在使用 UNIX 命令 diff 时,我得到几乎所有包含浮动值的行都不同!那是我的问题:我怎样才能得到像 'diff -u' 提供的结果差异,但对浮动比较的限制较少?

我想我会写一个 Python 的脚本来做到这一点,并找到了模块 difflib,它提供了类似 diff 的比较。但是我找到的文档解释了如何按原样使用它(通过单一方法),并解释了内部对象,但我找不到任何关于如何自定义 difflib 对象以满足我的需求的信息(比如只重写比较方法或这样)......我想一个解决方案可能是检索统一的差异,并“手动”解析它以消除我的“错误”差异,这并不优雅;我更喜欢使用已经存在的框架。

那么,有人知道如何定制这个库,以便我可以做我想要的吗?或者至少为我指明了正确的方向......如果不是在 Python 中,也许 shell 脚本可以完成这项工作?

任何帮助将不胜感激!提前感谢您的回答!

4

1 回答 1

4

在您的情况下,我们专门针对一般情况:在将内容传递到 difflib 之前,我们需要检测并单独处理包含浮点数的行。这是一种基本方法,如果您想生成增量、上下文行等,您可以在此基础上进行构建。请注意,将浮点数作为实际浮点数而不是字符串进行模糊比较更容易(尽管您可以逐列编写不同的代码,并忽略 1-e4 之后的字符)。

import re

float_pat = re.compile('([+-]?\d*\.\d*)')
def fuzzydiffer(line1,line2):
    """Perform fuzzy-diff on floats, else normal diff."""
    floats1 = float_pat.findall(line1)
    if not floats1:
        pass # run your usual diff() 
    else:
        floats2 = float_pat.findall(line2)
        for (f1,f2) in zip(floats1,floats2):
            (col1,col2) = line1.index(f1),line2.index(f2)
            if not fuzzy_float_cmp(f1,f2):
                print "Lines mismatch at col %d", col1, line1, line2
            continue
    # or use a list comprehension like all(fuzzy_float_cmp(f1,f2) for f1,f2 in zip(float_pat.findall(line1),float_pat.findall(line2)))
    #return match

def fuzzy_float_cmp(f1,f2,epsilon=1e-4):
    """Fuzzy-compare two strings representing floats."""
    float1,float2 = float(f1),float(f2)
    return (abs(float1-float2) < epsilon)

一些测试:

fuzzydiffer('text: 558.113509766 +23477547.6407 -0.867086648057 0.009291785451', 
'text: 558.11351 +23477547.6406 -0.86708665 0.009292000001')

作为奖励,这是一个突出列差异的版本:

import re

float_pat = re.compile('([+-]?\d*\.\d*)')
def fuzzydiffer(line1,line2):
    """Perform fuzzy-diff on floats, else normal diff."""
    floats1 = float_pat.findall(line1)
    if not floats1:
        pass # run your usual diff() 
    else:
        match = True
        coldiffs1 = ' '*len(line1)
        coldiffs2 = ' '*len(line2)
        floats2 = float_pat.findall(line2)
        for (f1,f2) in zip(floats1,floats2):
            (col1s,col2s) = line1.index(f1),line2.index(f2)
            col1e = col1s + len(f1)
            col2e = col2s + len(f2)
            if not fuzzy_float_cmp(f1,f2):
                match = False
                #print 'Lines mismatch:'
                coldiffs1 = coldiffs1[:col1s] + ('v'*len(f1)) + coldiffs1[col1e:]
                coldiffs2 = coldiffs2[:col2s] + ('^'*len(f2)) + coldiffs2[col2e:]
            #continue # if you only need to highlight first mismatch
        if not match:
            print 'Lines mismatch:'
            print '  ', coldiffs1
            print '< ', line1
            print '> ', line2
            print '  ', coldiffs2
        # or use a list comprehension like
        #    all()
        #return True

def fuzzy_float_cmp(f1,f2,epsilon=1e-4):
    """Fuzzy-compare two strings representing floats."""
    print "Comparing:", f1, f2
    float1,float2 = float(f1),float(f2)
    return (abs(float1-float2) < epsilon)
于 2011-07-03T01:07:15.680 回答