0

我有一个 python 代码，如果它们在反转时相似，它会删除行。例如，如果我有一个包含以下内容的文档：

1,2 3,4
5,6 7,8
2,1 4,3
5,6 8,7

执行脚本后，输出为

5,6 7,8
2,1 4,3
5,6 8,7

考虑第一列是 1,2，第二列是 7,8，然后，如果另一行包含每列的反转值 2,1 和 8,7，这被认为是反转的。

但是，我注意到脚本没有保持行的顺序。线路顺序对我来说很重要。另外，我需要删除第二个类似的反向行，而不是第一个。代码是

import sys

with open(sys.argv[1]) as inf:

    keys = set()

    for line in inf:

        ports, ips = line.split()

        port1, port2 = ports.split(",")

        ip1, ip2 = ips.split(",")

        if ip1 < ip2:

            keys.add((ip1, port1, ip2, port2))

        else:

            keys.add((ip2, port2, ip1, port1))

with open('results', 'w') as outf:

    for result in keys:

        outf.write("{1},{3}\t{0},{2}\n".format(*result))

有任何想法吗？如果我们可以在 bash 脚本上做到这一点，有什么建议吗？

谢谢

4

2 回答 2

2

你可以collections.OrderedDict在这里使用：

>>> from collections import OrderedDict
>>> dic = OrderedDict()
with open('file.txt') as f:
    for line in f:
        key = tuple(tuple(x.split(',')) for x in line.split())
        rev_key = tuple(x[::-1] for x in key)
        if key not in dic and rev_key not in dic:
            dic[key] = line.strip()
...             
>>> for v in dic.itervalues():
    print v
...     
1,2 3,4
5,6 7,8
5,6 8,7
于 2013-07-12T02:30:20.467 回答
1

既然你提到bash了，这是一个awk解决方案

awk -F'[ ,]' 'BEGIN{OFS=","} {$1=$1};
!($0 in arr){print($1,$2" "$3,$4);arr[$2","$1","$4","$3]}' file.txt

1,2 3,4
5,6 7,8
5,6 8,7
于 2013-07-12T03:44:49.057 回答