Python小记-- 读取当前目录下所有文件名

# -*- coding: utf-8 -*-
import os
def file_name(file_dir):
    with open("SelectAllFiles.txt","w") as f:
        for root, dirs, files in os.walk(file_dir,topdown=True):
            for name in files:
                f.write(name+'\n')
if __name__=="__main__":
    file_name(os.getcwd())