os.walk, не копаясь в каталогах ниже

Используйте функцию walklevel.

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[:]

Он работает так же, как 9X_python os.walk, но вы можете передать ему параметр level, который 9X_files указывает, насколько глубока рекурсия.

Думаю, решение на самом деле очень простое.

использовать

break

чтобы 9X_python-interpreter выполнить только первую итерацию цикла for, должен 9X_python быть более элегантный способ.

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

В первый раз, когда 9X_os.walk вы вызываете os.walk, он возвращает тюльпаны 9X_pythonista для текущего каталога, а затем в следующем 9X_py цикле содержимое следующего каталога.

Возьмите 9X_python исходный сценарий и просто добавьте перерыв.

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

Хорошее предложение использовать listdir. Прямой 9X_pythonic ответ на ваш вопрос в Python 2: root, dirs, files = os.walk(dir_name).next().

Эквивалентный 9X_pythonista синтаксис Python 3: root, dirs, files = next(os.walk(dir_name))

Вы можете использовать os.listdir(), который возвращает 9X_python список имен (как для файлов, так и для каталогов) в 9X_pythonista заданном каталоге. Если вам нужно различать 9X_os.walk файлы и каталоги, вызывайте os.stat() для каждого 9X_python имени.

Если у вас есть более сложные требования, чем 9X_pythonista только верхний каталог (например, игнорировать 9X_files каталоги VCS и т. д.), вы также можете изменить 9X_python список каталогов, чтобы предотвратить повторное 9X_os.walk прохождение через них os.walk.

т.е.

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()

Примечание. Будьте 9X_python-shell осторожны, чтобы изменить список, а не просто 9X_python перепривязать его. Очевидно, что os.walk 9X_pythonista не знает о внешнем перепривязке.

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

Мне захотелось бросить свои 2 пенса.

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

9X_python

Та же идея с listdir, но короче:

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

9X_pythonista

Начиная с Python 3.5, вы можете использовать os.scandir вместо 9X_pythonista os.listdir. Вместо строк вы получаете взамен итератор 9X_files объектов DirEntry. Из документов:

Использование scandir() вместо 9X_os.walk listdir() может значительно повысить производительность 9X_files кода, которому также требуется информация 9X_file о типе файла или атрибуте файла, поскольку 9X_os.walk объекты DirEntry предоставляют эту информацию, если 9X_python-interpreter операционная система предоставляет ее при 9X_python сканировании каталога. Все методы DirEntry могут 9X_pythonic выполнять системный вызов, но is_dir() и is_file() обычно 9X_python-shell требуют системного вызова только для символьных 9X_os.walk ссылок; DirEntry.stat() всегда требует системного вызова 9X_py в Unix, но требует только одного для символьных 9X_file ссылок в Windows.

Вы можете получить доступ 9X_pythonista к имени объекта через DirEntry.name, которое в этом случае 9X_pythonista эквивалентно выводу os.listdir

Вы также можете сделать следующее:

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

9X_file

В Python 3 мне удалось это сделать:

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] )

9X_pythonista

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
9X_py