os.walk mà không cần đào sâu vào các thư mục bên dưới

Làm cách nào để giới hạn os.walkchỉ trả lại các tệp trong thư mục mà tôi cung cấp?

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        for f in files:
            if os.path.splitext(f)[1] in whitelist:
                outputList.append(os.path.join(root, f))
            else:
                self._email_to_("ignore")
    return outputList

Sử dụng walklevelchức năng.

import os

def walklevel(some_dir, level=1):
    some_dir = some_dir.rstrip(os.path.sep)
    assert os.path.isdir(some_dir)
    num_sep = some_dir.count(os.path.sep)
    for root, dirs, files in os.walk(some_dir):
        yield root, dirs, files
        num_sep_this = root.count(os.path.sep)
        if num_sep + level <= num_sep_this:
            del dirs[:]

Nó hoạt động giống như vậy os.walk, nhưng bạn có thể chuyển cho nó một leveltham số cho biết mức độ sâu của đệ quy.

Không sử dụng os.walk.

Thí dụ:

import os

root = "C:\\"
for item in os.listdir(root):
    if os.path.isfile(os.path.join(root, item)):
        print item

Tôi nghĩ rằng giải pháp thực sự rất đơn giản.

sử dụng

break

để chỉ thực hiện lần lặp đầu tiên của vòng lặp for, phải có một cách thanh lịch hơn.

for root, dirs, files in os.walk(dir_name):
    for f in files:
        ...
        ...
    break
...

Lần đầu tiên bạn gọi os.walk, nó trả về hoa tulip cho thư mục hiện tại, sau đó trong vòng lặp tiếp theo nội dung của thư mục tiếp theo.

Lấy kịch bản gốc và chỉ cần thêm một đoạn nghỉ .

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        for f in files:
            if os.path.splitext(f)[1] in whitelist:
                outputList.append(os.path.join(root, f))
            else:
                self._email_to_("ignore")
        break
    return outputList

Đề xuất để sử dụng listdirlà một trong những tốt. Câu trả lời trực tiếp cho câu hỏi của bạn trong Python 2 là root, dirs, files = os.walk(dir_name).next().

Cú pháp Python 3 tương đương là root, dirs, files = next(os.walk(dir_name))

Bạn có thể sử dụng os.listdir()nó trả về danh sách các tên (cho cả tệp và thư mục) trong một thư mục nhất định. Nếu bạn cần phân biệt giữa tệp và thư mục, hãy gọi os.stat()từng tên.

Nếu bạn có nhiều yêu cầu phức tạp hơn chỉ là thư mục trên cùng (ví dụ: bỏ qua các dirs VCS, v.v.), bạn cũng có thể sửa đổi danh sách các thư mục để ngăn os.walk đệ quy qua chúng.

I E:

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        dirs[:] = [d for d in dirs if is_good(d)]
        for f in files:
            do_stuff()

Lưu ý - hãy cẩn thận thay đổi danh sách, thay vì chỉ gắn lại nó. Rõ ràng là os.walk không biết về sự ràng buộc bên ngoài.

for path, dirs, files in os.walk('.'):
    print path, dirs, files
    del dirs[:] # go only one level deep

Ý tưởng tương tự với listdir, nhưng ngắn hơn:

[f for f in os.listdir(root_dir) if os.path.isfile(os.path.join(root_dir, f))]

Cảm giác như ném 2 pence của tôi vào.

baselevel = len(rootdir.split("\\"))
for subdirs, dirs, files in os.walk(rootdir):
    curlevel = len(subdirs.split("\\"))
    if curlevel <= baselevel + 1:
        [do stuff]

Trong Python 3, tôi đã có thể làm điều này:

import os
dir = "/path/to/files/"

#List all files immediately under this folder:
print ( next( os.walk(dir) )[2] )

#List all folders immediately under this folder:
print ( next( os.walk(dir) )[1] )

Kể từ Python 3.5, bạn có thể sử dụng os.scandirthay thế os.listdir. Thay vì các chuỗi, bạn nhận được một trình lặp lại các DirEntryđối tượng. Từ các tài liệu:

Việc sử dụng scandir()thay vì listdir()có thể làm tăng đáng kể hiệu suất của mã cũng cần loại tệp hoặc thông tin thuộc tính tệp, vì DirEntrycác đối tượng tiết lộ thông tin này nếu hệ điều hành cung cấp thông tin đó khi quét thư mục. Tất cả các DirEntryphương thức có thể thực hiện một lệnh gọi hệ thống, nhưng is_dir()và is_file()thường chỉ yêu cầu một lệnh gọi hệ thống cho các liên kết tượng trưng; DirEntry.stat()luôn yêu cầu lệnh gọi hệ thống trên Unix nhưng chỉ yêu cầu một lệnh gọi cho các liên kết tượng trưng trên Windows.

Bạn có thể truy cập vào tên của đối tượng DirEntry.namemà qua đó tương đương với đầu ra củaos.listdir

Bạn cũng có thể làm như sau:

for path, subdirs, files in os.walk(dir_name):
    for name in files:
        if path == ".": #this will filter the files in the current directory
             #code here

Đây là cách tôi đã giải quyết nó

if recursive:
    items = os.walk(target_directory)
else:
    items = [next(os.walk(target_directory))]

...

Có một vấn đề khi sử dụng listdir. Os.path.isdir (định danh) phải là một đường dẫn tuyệt đối. Để chọn các thư mục con, bạn thực hiện:

for dirname in os.listdir(rootdir):
  if os.path.isdir(os.path.join(rootdir, dirname)):
     print("I got a subdirectory: %s" % dirname)

Cách thay thế là thay đổi thư mục để thực hiện kiểm tra mà không có os.path.join ().

Bạn có thể sử dụng đoạn mã này

for root, dirs, files in os.walk(directory):
    if level > 0:
        # do some stuff
    else:
        break
    level-=1

tạo danh sách các loại trừ, sử dụng fnmatch để bỏ qua cấu trúc thư mục và thực hiện quy trình

excludes= ['a\*\b', 'c\d\e']
for root, directories, files in os.walk('Start_Folder'):
    if not any(fnmatch.fnmatch(nf_root, pattern) for pattern in excludes):
        for root, directories, files in os.walk(nf_root):
            ....
            do the process
            ....

giống như đối với 'bao gồm':

if **any**(fnmatch.fnmatch(nf_root, pattern) for pattern in **includes**):

Tại sao không chỉ đơn giản sử dụng a rangevà os.walkkết hợp với zip? Không phải là giải pháp tốt nhất, nhưng cũng sẽ hoạt động.

Ví dụ như thế này:

# your part before
for count, (root, dirs, files) in zip(range(0, 1), os.walk(dir_name)):
    # logic stuff
# your later part

Làm việc cho tôi trên python 3.

Ngoài ra: A breakcũng đơn giản hơn btw. (Xem câu trả lời từ @Pieter)

Một chút thay đổi đối với câu trả lời của Alex, nhưng sử dụng __next__():

print(next(os.walk('d:/'))[2]) hoặc là print(os.walk('d:/').__next__()[2])

với [2]là filetrong root, dirs, fileđề cập trong câu trả lời khác

thư mục gốc thay đổi cho mọi thư mục mà os.walk tìm thấy. Tôi giải quyết rằng kiểm tra xem thư mục root ==

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        if root == dir_name: #This only meet parent folder
            for f in files:
                if os.path.splitext(f)[1] in whitelist:
                    outputList.append(os.path.join(root, f))
                else:
                    self._email_to_("ignore")
    return outputList

import os

def listFiles(self, dir_name):
    names = []
    for root, directory, files in os.walk(dir_name):
        if root == dir_name:
            for name in files:
                names.append(name)
    return names