利用python列出当前目录下的所有文件

问题

当一个目录下有很多文件夹或者文件，我们想分析各个文件的名字，这时就可以写一个函数，列出当前目录下所有文件名字。

代码

src_dir = r'./'      # 源文件目录地址

def list_all_files(rootdir):
    import os
    _files = []

    #列出文件夹下所有的目录与文件
    list_file = os.listdir(rootdir)

    for i in range(0,len(list_file)):

        # 构造路径
        path = os.path.join(rootdir,list_file[i])

        # 判断路径是否是一个文件目录或者文件
        # 如果是文件目录，继续递归

        if os.path.isdir(path):
            _files.extend(list_all_files(path))
        if os.path.isfile(path):
             _files.append(path)
    return _files

files = list_all_files(src_dir)
print(files)