Làm cách nào để lặp lại các tệp trong một thư mục nhất định?

Question

Tôi cần phải lặp qua tất cả các tệp .asm bên trong một thư mục nhất định và thực hiện một số thao tác trên chúng.

Làm cách nào để thực hiện điều này một cách hiệu quả?

Answer 1

Original câu trả lời:

for filename in os.listdir(directory): 
    if filename.endswith(".asm") or filename.endswith(".py"): 
     # print(os.path.join(directory, filename)) 
     continue 
    else: 
     continue 

Python 3.6 phiên bản của câu trả lời ở trên, sử dụng os - giả định rằng bạn có đường dẫn thư mục như một đối tượng str trong một biến gọi là directory_in_str:

directory = os.fsencode(directory_in_str) 

for file in os.listdir(directory): 
    filename = os.fsdecode(file) 
    if filename.endswith(".asm") or filename.endswith(".py"): 
     # print(os.path.join(directory, filename)) 
     continue 
    else: 
     continue 

Hoặc đệ quy, sử dụng pathlib:

from pathlib import Path 

pathlist = Path(directory_in_str).glob('**/*.asm') 
for path in pathlist: 
    # because path is object not string 
    path_in_str = str(path) 
    # print(path_in_str) 

Answer 2

Bạn có thể thử sử dụng glob mô-đun

import glob 

for filename in glob.iglob('/foobar/*.asm'): 
    print('/foobar/%s' % filename) 

Answer 3

này sẽ lặp qua tất cả các file hậu duệ, không chỉ là những đứa trẻ trước mắt của các thư mục:

import os 

for subdir, dirs, files in os.walk(rootdir): 
    for file in files: 
     #print os.path.join(subdir, file) 
     filepath = subdir + os.sep + file 

     if filepath.endswith(".asm"): 
      print (filepath) 

Answer 4

Python 3.4 và sau đó cung cấp pathlib trong thư viện chuẩn. Bạn có thể làm:

from pathlib import Path 

asm_pths = [pth for pth in Path.cwd().iterdir() 
      if pth.suffix == '.asm'] 

Hoặc nếu bạn không thích comprehensions danh sách:

asm_paths = [] 
for pth in Path.cwd().iterdir(): 
    if pth.suffix == '.asm': 
     asm_pths.append(pth) 

Path đối tượng có thể dễ dàng được chuyển đổi sang chuỗi.

Answer 5

Tôi chưa hoàn toàn hài lòng với việc triển khai này, tôi muốn có một hàm tạo tùy chỉnh thực hiện DirectoryIndex._make(next(os.walk(input_path))) để bạn có thể chuyển đường dẫn mà bạn muốn danh sách tệp. Chỉnh sửa chào mừng!

import collections 
import os 

DirectoryIndex = collections.namedtuple('DirectoryIndex', ['root', 'dirs', 'files']) 

for file_name in DirectoryIndex(*next(os.walk('.'))).files: 
    file_path = os.path.join(path, file_name)