python獲取指定目錄下所有文件名列表的方法？

寫(xiě)段偽代碼import os# 獲取當(dāng)前目錄path = os.getcwd()# 獲取當(dāng)前目錄所有文件列表list_dir = os.listdir(path)for i in list_dir: # 字符串分割，判斷"."后面是否為"py" if i.split(".")[-1] == "py": print(i)