如何關閉此功能中使用的文件？

Question

我有以下功能。該程序查看每個文件並將所有4個文件中出現的行打印到新文件中。我試過file1.close()，但是我收到關於關閉一個集合的錯誤？我想我可以使用with聲明，但不知道如何做到這一點，我對編程非常陌生。

def secretome(): 
    file1 = set(line.strip() for line in open(path + "goodlistSigP.txt")) 
    file2 = set(line.strip() for line in open(path + "tmhmmGoodlist.txt")) 
    file3 = set(line.strip() for line in open(path + "targetpGoodlist.txt")) 
    file4 = set(line.strip() for line in open(path + "wolfPsortGoodlist.txt")) 
    newfile = open(path + "secretome_pass.txt", "w") 
    for line in file1 & file2 & file3 & file4: 
     if line: 
      newfile.write(line + '\n') 
    newfile.close() 

Answer 1

我建議通過抽取您所設定的代入功能刪除重複：

def set_from_file(path): 
    with open(path) as file: 
     return set(lines.strip() for line in file) 

def secretome(): 
    files = ["goodlistSigP.txt", "tmhmmGoodlist.txt", "targetpGoodlist.txt", "wolfPsortGoodlist.txt"] 
    data = [set_from_file(os.path.join(path, file)) for file in files] 
    with open(path + "secretome_pass.text", "w") as newfile: 
     newfile.writelines(line + "/n" for line in set.union(*data) if line) 

注意，你在你的代碼做交集，但你說說想要一個工會，所以我用union()在這裏。還有一些list comprehensions/generator expressions。

Answer 2

你可以把它變成一個發電機：

def closingfilelines(*a): 
    with open(*a) as f: 
     for line in f: 
      yield f 

，並使用它，你目前使用的open()。

生成器運行時，文件保持打開狀態，如果生成器耗盡，它將被關閉。

同樣的情況，如果發電機對象是.close() d或下降 - 在這種情況下，發電機得到一個GeneratorExit例外，這使得with子句離開爲好。

Answer 3

藉此完全不同的方向上比我原來的（這Lattyware打我）：

你可以定義一個函數：

def file_lines(fname): 
    with open(fname) as f: 
     for line in f: 
      yield line 

現在你可以使用itertools.chain遍歷文件：

import itertools 
def set_from_file(path): 
    filenames = ("name1","name2","name3",...) #your input files go here 
    lines = itertools.chain.from_iterable(itertools.imap(file_lines,filenames)) 
    #lines is an iterable object. 
    #At this point, virtually none of your system's resources have been consumed 
    with open("output",'w') as fout: 
     #Now we only need enough memory to store the non-duplicate lines :) 
     fout.writelines(set(line.strip()+'\n' for line in lines)) 

Answer 4

這似乎是一個非常複雜的做法。我會建議像我在這裏給出的例子。

import fileinput 
files = ['file1.txt','file2.txt','file3.txt','file4.txt'] 
output = open('output.txt','w') 

for file in files: 
    for line in fileinput.input([file]): 
     output.write(line) 
    output.write('\n') 

output.close() 

此代碼創建與它的文件（更換所需的文件路徑名）的列表，創建一個文件來存儲每個輸出，然後簡單地通過他們迭代使用的FileInput模塊要經過各一行一行地逐行打印每行到輸出文件。 'output.write（'\ n'）'確保下一個文件行的打印從輸出文件的新行開始。