使用os.listdir() 函數來獲取某個目錄中的文件列表
import os names = os.listdir('somedir')
結果會返回目錄中所有文件列表,包括所有文件,子目錄,符號鏈接等等。如果你需要通過某種方式過濾數據,可以考慮結合os.path 庫中的一些函數來使用列表推導
names = [name for name in os.listdir('somedir')if os.path.isfile(os.path.join('somedir', name))]
字符串的startswith() 和endswith() 方法對於過濾一個目錄的內容也是很有用的。
pyfiles = [name for name in os.listdir('somedir')if name.endswith('.py')]