Question

我有两个制表符分隔的文件，我需要针对另一个文件中的所有行测试第一个文件中的每一行。例如，

文件1：

row1    c1    36    345   A
row2    c3    36    9949  B
row3    c4    36    858   C

file2的：

row1    c1    3455  3800
row2    c3    6784  7843
row3    c3    10564 99302
row4    c5    1405  1563

假设我想输出（file1）中的所有行，其中file1的col [3]小于file2的任何（不是每个）col [2]，因为col [1]是相同的。

预期产出：

row1    c1    36    345   A
row2    c3    36    9949  B

由于我在Ubuntu工作，我希望输入命令看起来像这样：
python code.py [file1] [file2] > [output]

我写了以下代码：

import sys

filename1 = sys.argv[1]
filename2 = sys.argv[2]

file1 = open(filename1, 'r')
file2 = open(filename2, 'r')

done = False

for x in file1.readlines():
    col = x.strip().split()
    for y in file2.readlines():
        col2 = y.strip().split()
        if col[1] == col2[1] and col[3] < col2[2]:
            done = True
            break
        else: continue
print x

但是，输出如下所示：

row2    c3    36    9949  B

这对于较大的数据集很明显，但基本上我总是只得到嵌套循环中的条件为真的最后一行。我怀疑“休息”让我脱离了两个循环。我想知道（1）如何打破只有一个for循环，以及（2）如果这是我遇到的唯一问题。

Answer 1

break和continue适用于最里面的循环。

问题是您只打开第二个文件一次，因此只读一次。当您第二次执行for y in file2.readlines():时，file2.readlines()将返回一个空的可迭代。

将file2 = open(filename2, 'r')移至外圈，或使用seek()快退至file2的开头。

Answer 2

您需要将数字字符串解析为相应的整数值。

您可以按如下方式使用int('hoge')。

import sys

filename1 = sys.argv[1]
filename2 = sys.argv[2]

with open(filename1) as file1:
    for x in file1:
        with open(filename2) as file2:
            col = x.strip().split()
            for y in file2:
                col2 = y.strip().split()
                if col[1] == col2[1] and int(col[3]) < int(col2[2]):
                    print x

如何打破只有一个嵌套循环

2 个答案: