Không đệ quy os.walk()

Question

Tôi đang tìm cách để thực hiện một chuyến đi không đệ quy os.walk(), giống như os.listdir() hoạt động. Nhưng tôi cần phải quay trở lại giống như cách trả về os.walk(). Bất kỳ ý tưởng?

Cảm ơn bạn trước.

Answer 1

next(os.walk(...)) 

Answer 2

My giải pháp parametrised hơn một chút sẽ là:

for root, dirs, files in os.walk(path): 
    if not recursive: 
     while len(dirs) > 0: 
      dirs.pop() 

    //some fency code here using generated list 

Edit: sửa lỗi, nếu/khi vấn đề. Cảm ơn, @Dirk van Oosterbosch:}

Answer 3

Vâng những gì Kamiccolo nghĩa là phù hợp hơn với điều này:

for str_dirname, lst_subdirs, lst_files in os.walk(str_path): 
    if not bol_recursive: 
      while len(lst_subdirs) > 0: 
       lst_subdirs.pop() 

Answer 4

Thêm một break sau tên tập tin cho vòng lặp:

for root, dirs, filenames in os.walk(workdir): 
    for fileName in filenames: 
     print (fileName) 
    break #prevent decending into subfolders 

này hoạt động vì (theo mặc định) os.walk đầu tiên liệt kê các tệp trong thư mục được yêu cầu và sau đó đi vào các thư mục con.