Bạn có thể sử dụng phương thức os.listdir để lấy tất cả các thư mục và tệp trong một thư mục. Sau đó, lọc danh sách để chỉ lấy các tệp và kiểm tra cả phần mở rộng của chúng.
Ví dụ
>>> import os
>>> file_list = [f for f in os.listdir('.') if os.path.isfile(os.path.join('.', f)) and f.endswith('.txt')]
>>> print file_list
['LICENSE.txt', 'NEWS.txt', 'README.txt'] Phương thức endwith là một thành viên của lớp chuỗi kiểm tra xem một chuỗi có kết thúc bằng một hậu tố nhất định hay không.
Bạn cũng có thể sử dụng mô-đun cầu để đạt được điều tương tự:
>>> import glob, os
>>> file_list = [f for f in glob.glob("*.txt")]
>>> print file_list
['LICENSE.txt', 'NEWS.txt', 'README.txt']