python學習——把讀取fasta文件的代碼封裝成函數

將讀取文件的代碼封裝成函數，並使其作為模塊可在其他程序運行

創建fasta_def.py文件，並輸入如下代碼：

def read_fasta(input): #用def定義函數read_fasta()，並向函數傳遞參數用變量input接收
    with open(input,'r') as f: # 打開文件
        fasta = {} # 定義一個空的字典
        for line in f:
            line = line.strip() # 去除末尾換行符
            if line[0] == '>':
                header = line[1:]
            else:
                sequence = line
                fasta[header] = fasta.get(header,'') + sequence
    return fasta 


if __name__ == '__main__': # 函數若是在當前程序運行則執行下面的代碼，若是在另外的程序作為模塊運行則不執行下面的代碼
    fasta_info = read_fasta('genome_test.fa')
    print(fasta_info)

新建一個test.py文件

輸入如下代碼調用上述模塊fasta_def

import fasta_def

fa = fasta_def.read_fasta('genome_test.fa')