python - 如何输入具有特定结尾的目录中的所有文件？Python

Question

我有一个名为 Test 的文件夹'/Desktop/Test/'

我的文件夹中有几个文件（例如 1.fa、2.fa、3.fa、X.fa）

或者'/Desktop/Test/1.fa','/Desktop/Test/2.fa','/Desktop/Test/3.fa','/Desktop/Test/X.fa'

我正在尝试在我的函数中创建一个元素，以打开目录 '/Desktop/Test/'中以.fa结尾的每个文件，而不用实际创建一个包含 20 个左右变量的函数（这是我知道如何做的唯一方法）

例子：

def simple(input):

    #input would be the directory '/Desktop/Test/'
    #for each .fa in the directory (eg. 1.fa, 2.fa, 3.fa, X.fa) i need it to create a list of all the strings within each file 
    #query=open(input,'r').read().split('\n') is what I would use in my simple codes that only have one input

如何输入具有特定结尾（.fa）的目录中的所有文件？

Answer 1

你可以做：

import glob
import os
def simple(input)
    os.chdir(input)
    for file in glob.glob("*.fa"):
        with open(file, 'r+') as f:
            print f.readlines() #or do whatever