Python如何获取当前文件夹下所有文件名

Python如何获取当前文件夹下所有文件名

最近遇到一个问题，需要统计某文件夹下面的子文件夹标题信息，一个一个统计速度太慢，就打算用python来实现。放到此处，以备后用。直接放代码，具体如下：

import os
import sys


def file_name(file_dir):  
    for root, dirs, files in os.walk(file_dir):   
        #备注root返回当前目录路径；dirs返回当前路径下所有子目录；files返回当前路径下所有非目录子文件
        return root,dirs,files

if __name__=='__main__':
    file_dir = input("请输入目录（注意用斜杠/隔开，按ENTER键退出）：")  
    
    if file_dir == "":
         sys.exit(0)
   
    save_filename = input("请输入要保存的文件名：")+".txt"
    
    root,dirs,files = file_name(file_dir)
    f = open(save_filename, 'w',encoding = 'utf-8')
    for line in dirs:  #如果要写入所有的非目录子文件（如.docx,.xlxs文件名，则此处将dirs替换成files变量）
        f.write(line+'\n')
    f.close()
    
    print("文件\'{}\'已保存成功！".format(save_filename))