利用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)