菜鳥學python之用python找指定檔案

編寫python程式查詢某個路徑下的檔案

import os
#指定某個路徑
path = "/home/aistudio/Day1-homework"
def walk_ext_file(dir_path,ext):
    # 遍歷
    for root, dirs, files in os.walk(dir_path):
        # 獲取檔名稱及路徑
        for file in files:
            file_path = os.path.join(root, file)
            
            file_name=os.path.basename(file_path)
            value=file_name.find(ext)
            
            if  value != -1:
            #找到之後列印
                print(file_path)
if __name__ == '__main__':
#查詢包含2020的檔案：
	walk_ext_file(path,'2020')

結果：
/home/aistudio/Day1-homework/18/182020.doc
/home/aistudio/Day1-homework/4/22/04:22:2020.txt
/home/aistudio/Day1-homework/26/26/new2020.txt

這是百度飛漿課程的作業題，沒看老師答案，我自己完成的哦。