从python中删除文件中的行

时间:2015-07-01 23:37:42

标签: python file-io file-processing

我的文件格式如下:

#1  0.13297254902 0.324803921569 0.434835294118 ...#many floats in one line
#2  0
#3  0.377305882353 0.595870588235 0.353215686275 ...
#4  1 #0/1 for the second line 
#5  ....

我想处理文件,以便删除第二行的所有块都为0,并将文件保留为

#1  0.377305882353 0.595870588235 0.353215686275 ...
#2  1
#3  0.403529411765 0.341654901961 0.379278431373 ... 
#4  1 #now they are all 1s
#5  .....

我尝试了下面的代码片段,但它只能看到0/1然后删除该行,但是我想删除0/1以上浮点数的行,而不是0/1以下的浮点数行。

f = open(filePath, "r")
lines = f.readlines()
f.close()
f = open(filePath, "w")

for line in lines:
    if "1\n" in line:
        f.write(line)

有没有其他方法可以选择要包含哪一行而哪些不包含? 或者可能有一种方法可以向后处理文件?

1 个答案:

答案 0 :(得分:2)

我们可以使用next()函数来获取文件中的下一个元素iterable。 shutil模块允许我们移动新文件,覆盖原文(感谢@JoranBeasley)。

import shutil

with open(filePath, 'r') as f, open('new_' + filePath, 'w') as output:
    for line in f:
        n = next(f)
        if n != '0\n':
            output.write(line+n)

shutil.move("new_" + filePath, filePath)

输入:

0.13297254902 0.324803921569 0.434835294118 ...#many floats in one line
0
0.377305882353 0.595870588235 0.353215686275 ...
1 #0/1 for the second line

输出:

0.377305882353 0.595870588235 0.353215686275 ...
1 #0/1 for the second line