Python: os.listdir thay thế/một số tiện ích mở rộng

Question

Có thể xem các tệp có phần mở rộng nhất định bằng lệnh os.listdir không? Tôi muốn nó hoạt động để nó chỉ hiển thị các tệp hoặc thư mục có đuôi .f ở cuối. Tôi đã kiểm tra tài liệu và không tìm thấy gì cả, vì vậy đừng hỏi.

Answer 1

glob là tốt tại đây:

import glob 
for f in glob.glob("*.f"): 
    print f 

Answer 2

Đừng hỏi gì?

[s for s in os.listdir() if s.endswith('.f')] 

Nếu bạn muốn kiểm tra một danh sách các phần mở rộng, bạn có thể làm cho khái quát rõ ràng,

[s for s in os.listdir() if s.endswith('.f') or s.endswith('.c') or s.endswith('.z')] 

hay cách khác đây là ngắn hơn một chút để viết:

[s for s in os.listdir() if s.rpartition('.')[2] in ('f','c','z')] 

Answer 3

[s for s in os.listdir() if os.path.splitext(s) == 'f'] 

Answer 4

Có một khả năng không được đề cập cho đến nay:

import fnmatch 
import os 

for file in os.listdir('.'): 
    if fnmatch.fnmatch(file, '*.f'): 
     print file 

Trên thực tế đây là cách các module glob được thực hiện, vì vậy trong trường hợp này glob là đơn giản hơn và tốt hơn, nhưng các module fnmatch có thể tiện dụng trong khác tình huống, ví dụ khi thực hiện truyền tải cây bằng cách sử dụng os.walk.

Answer 5

Hãy thử điều này:

from os import listdir 

extension = '.wantedExtension' 

mypath = r'my\path' 

filesWithExtension = [ f for f in listdir(mypath) if f[(len(f) - len(extension)):len(f)].find(extension)>=0 ]