批量读取路径下的文件名并保存在txt文件里

import os

def ListFilesToTxt(dir, file, wildcard, recursion):
    exts = wildcard.split(" ")
    files = os.listdir(dir)
    for name in files:
        fullname = os.path.join(dir, name)
        if (os.path.isdir(fullname) & recursion):
            ListFilesToTxt(fullname, file, wildcard, recursion)
        else:
            for ext in exts:
                if (name.endswith(ext)):
                    file.write(dir + "\\" + name + "\n")
                    break

def Test():
    dir = "D:\\datadoc"  # 文件路径
    outfile = "D:\\test.txt"  # 写入的txt文件名
    wildcard = ".nii"  # 以上文件路径下要读取的文件类型，比如nii格式
    file = open(outfile, "w")
    if not file:
        print("cannot open the file %s for writing" % outfile)
    ListFilesToTxt(dir, file, wildcard, 1)
    file.close()

Test()