输出始终给出最后一行或所有行

时间:2013-09-01 08:51:32

标签: python csv

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

文件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]> [输出]

我写了以下代码:

import sys

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

file1 = open(filename1, 'r')

done = False

for x in file1.readlines():
    col = x.strip().split()
    file2 = open(filename2, 'r')
    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  

基本上我总是得到嵌套循环中的条件为真的最后一行。我尝试了这个:

    if done == True: print x

(有一个缩进),但现在它打印file1中的所有行,无论前一个循环中测试的条件如何。 (大于_≤)

1 个答案:

答案 0 :(得分:3)

您忘记在第一次匹配后重置done变量,在这种情况下您不需要变量。要修复代码,只需将done = True替换为print x,然后使用int(col[3]) < int(col2[2])将列作为数字(整数)进行比较。