2

我正在尝试以某种方式从 python 中的文件中删除特定的行号，例如

./foo.py 文件名.txt 4 5 2919

其中 4 5 和 2919 是行号

我想做的是：

for i in range(len(sys.argv)):
    if i>1: # Avoiding sys.argv[0,1]
        newlist.append(int(sys.argv[i]))

然后：

count=0

generic_loop{ 
   bar=file.readline()
   count+=1
   if not count in newlist:
      print bar
}

它打印原始文件中的所有行（中间有空格）

4

4 回答 4

3

您可以使用enumerate来确定行号：

import sys
exclude = set(map(int, sys.argv[2:]))
with open(sys.argv[1]) as f:
    for num,line in enumerate(f, start=1):
        if num not in exclude:
            sys.stdout.write(line)

如果从 0 开始计数，则可以删除start=1。在上面的代码中，行号从 1 开始：

$ python3 so-linenumber.py so-linenumber.py 2 4 5
import sys
with open(sys.argv[1], 'r') as f:
            sys.stdout.write(line)

如果要将内容写入文件本身，请将其写入临时文件而不是 sys.stdout，然后将其重命名为原始文件名（或在命令行上使用海绵），如下所示：

import os
import sys
from tempfile import NamedTemporaryFile
exclude = set(map(int, sys.argv[2:]))
with NamedTemporaryFile('w', delete=False) as outf:
    with open(sys.argv[1]) as inf:
        outf.writelines(line for n,line in enumerate(inf, 1) if n not in exclude)
    os.rename(outf.name, sys.argv[1])
于 2013-05-06T02:16:03.983 回答
2

你可以尝试这样的事情：

import sys
import os
filename= sys.argv[1]
lines = [int(x) for x in sys.argv[2:]]

#open two files one for reading and one for writing

with open(filename) as f,open("newfile","w") as f2:

#use enumerate to get the line as well as line number, use enumerate(f,1) to start index from 1
    for i,line in enumerate(f):  
        if i not in lines:     #`if i not in lines` is more clear than `if not i in line`
            f2.write(line)   
os.rename("newfile",filename)  #rename the newfile to original one

请注意，对于临时文件的生成，最好使用tempfilemodule.

于 2013-05-06T02:17:18.803 回答
2 
import sys
# assumes line numbering starts with 1
# enumerate() starts with zero, so we subtract 1 from each line argument
omitlines = set(int(arg)-1 for arg in sys.argv[2:] if int(arg) > 0)
with open(sys.argv[1]) as fp:
    filteredlines = (line for n,line in enumerate(fp) if n not in omitlines)
    sys.stdout.writelines(filteredlines)
于 2013-05-06T02:21:37.050 回答
1

该fileinput模块有一个inplace=True选项可以将标准输出重定向到一个临时文件，该文件会自动为您重命名。

import fileinput
exclude = set(map(int, sys.argv[2:]))

for i, line in enumerate(fileinput.input('filename.txt', inplace=True), start=1):
    if i not in exclude:
        print line, # fileinput inplace=True redirects stdout to tempfile
于 2013-05-06T10:30:08.340 回答