питона удалить «многие» строку из файла

Question

Я пытаюсь удалить определенные номера строк из файла в Python в пути, такие как

./foo.py filename.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 
} 

печатает все строки в исходном файле (с пробелами между ними)

Answer 1

Вы можете попробовать что-то вроде этого:

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  

Обратите внимание, что для создания временных файлов, что лучше использовать tempfile модуль.

Answer 2

Вы можете использовать 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) 

Вы можете удалить start=1 при запуске считая в 0. В приведенном выше коде нумерация строк начинается с 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) 

Если вы хотите написать контент самому файлу, напишите его в temporary file вместо sys.stdout, а затем rename, чтобы исходное имя файла (или используйте sponge в командной строке), например:

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]) 

Answer 3

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) 

Answer 4

Модуль fileinput имеет inplace=True вариант, который перенаправляет стандартный вывод в TempFile который автоматически переименованной после вас.

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