对两个文本执行diff,只使用文本中每行的一部分

2024-05-28 20:02:40 发布

您现在位置:Python中文网/ 问答频道 /正文

我有两条短信。你知道吗

T0 ID A
T1 ID B
T2 ID C
T4 ID D

以及

T5 ID A
T6 ID E
T7 ID F
T8 ID D

我真正感兴趣的只是比较身份证的顺序,字母的不同。但是我想在输出中保留每个元素的第一部分,这样我可以在以后的其他分析中使用它

@-1,2 +1,2
-T2 ID B
-T3 ID C
+T6 ID E
+T7 ID F

做这种区别最好的方法是什么?最好是使用difflib python库。你知道吗

另外,假设不是2个文本,而是2个对象列表,每个对象都有2个属性(object.t返回T1,对象ID返回B),我只想对对象列表中的ID属性执行diff。我能做这样的手术吗?也许这应该是一个不同的问题。你知道吗

谢谢


Tags: 对象id列表属性短信感兴趣t1身份证
1条回答
网友
1楼 · 发布于 2024-05-28 20:02:40

您可以创建str的子类,对其进行散列和比较,就好像它只包含其ID:

import re

class IdString(str):
    """A string that hashes and compares on its id.

        >>> hash(IdString('XXX ID A XXX')) == hash('A')
        True
        >>> hash(IdString('XXX ID abc XXX')) == hash('abc')
        True
        >>> IdString('XXX ID A XXX') == IdString('YYY ID A YYY')
        True
        >>> IdString('XXX ID A XXX') == IdString('XXX ID B XXX')
        False

    """
    def __new__(cls, *args):
        self = super(IdString, cls).__new__(cls, *args)
        m = re.search(r'\bID (\w+)', self)
        self.id = m.group(1)
        return self

    def __hash__(self):
        return hash(self.id)

    def __eq__(self, other):
        return self.id == other.id

    def __ne__(self, other):
        return self.id != other.id

然后可以将普通字符串转换为IdString对象并将其传递给difflib,如下所示:

from difflib import unified_diff

text1 = '''T0 ID A
T1 ID B
T2 ID C
T4 ID D
'''

text2 = '''T5 ID A
T6 ID E
T7 ID F
T8 ID D
'''

print(''.join(unified_diff(map(IdString, text1.splitlines(True)),
                           map(IdString, text2.splitlines(True)),
                           n=0)))

产生非常接近你想要的输出:

 - 
+++ 
@@ -2,2 +2,2 @@
-T1 ID B
-T2 ID C
+T6 ID E
+T7 ID F

(你的问题中的例子是@-1,2 +1,2,但我无法准确地再现,因为我不知道diff是什么样式,在diff输出中行号从1开始。)

相关问题 更多 >

    热门问题