Python腳本來檢索文件從文件夾的路徑到一個文本文件

Question

import shutil 
import os 

def get_files(): 
    source = os.listdir("/output_folder/sample/cufflinks/") 
    destination = "output_folder/assemblies.txt" 
    for files in source: 
     if files.with("transcripts.gtf"): 
      shutil.move(files,destination) 

我要檢索transcripts.gtf文件從"/output_folder/sample/cufflinks/"到assemblies.txt。上面的代碼是否正確。請幫助我。謝謝 ！！

Answer 1

您可以使用os.walk：

import os 
import shutil 
from os.path import join 

destination = "/output_folder/assemblies.txt" # Absolute path, or use join 
folder_to_look_in = "/output_folder/sample/cufflinks/" # Absolute folder path, or use join 
for _, _, files in os.walk(folder_to_look_in): 
    for file_name in files: 
     if file_name.endswith("transcripts.gtf"): 
      try: 
       # 'a' will append the data to the tail of the file 
       with open(destination, 'a') as my_super_file: 
        my_super_file.write(file_name) 
      except OSError as e: 
       print "I/O error({0}): {1}".format(e.errno, e.strerror) 

Answer 2

我想，我終於明白你想達到的目標。使用glob.glob()功能：

import glob 
destination = "output_folder/assemblies.txt" 
source = '/output_folder/sample/cufflinks/*/*transcripts.gtf' 
with open(destination, 'w') as f: 
    f.write('\n'.join(glob.glob(source))+'\n') 

f.write('\n'.join(glob.glob(source))+'\n')是等價的：

s = '' 
for path in glob.glob(source): 
    s += '{}\n'.format(path) 
f.write(s)