아래 디렉토리를 파지 않고 os.walk

os.walk내가 제공 한 디렉토리의 파일 만 반환하도록 제한 하려면 어떻게합니까 ?

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

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

처럼 작동 os.walk하지만 level재귀의 깊이를 나타내는 매개 변수를 전달할 수 있습니다 .

os.walk를 사용하지 마십시오.

예:

import os

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

해결책은 실제로 매우 간단하다고 생각합니다.

사용하다

break

for 루프의 첫 번째 반복 만 수행하려면 더 우아한 방법이 있어야합니다.

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

os.walk를 처음 호출하면 현재 디렉토리에 대한 튤립을 반환 한 다음 다음 디렉토리의 내용을 루프합니다.

원본 스크립트를 가져 와서 휴식을 추가하십시오 .

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은 좋은 것입니다. Python 2에서 귀하의 질문에 대한 직접적인 대답은 root, dirs, files = os.walk(dir_name).next().

동등한 Python 3 구문은 다음과 같습니다. root, dirs, files = next(os.walk(dir_name))

os.listdir()주어진 디렉토리에서 이름 목록 (파일과 디렉토리 모두)을 반환하는 것을 사용할 수 있습니다 . 파일과 디렉토리를 구별해야하는 경우 os.stat()각 이름을 호출하십시오 .

최상위 디렉토리보다 더 복잡한 요구 사항이있는 경우 (예 : VCS 디렉토리 무시 등) 디렉토리 목록을 수정하여 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()

참고-목록을 리 바인드하는 것보다 변경하는 데주의하십시오. 분명히 os.walk는 외부 리 바인딩에 대해 알지 못합니다.

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

와 같은 아이디어 listdir이지만 더 짧습니다.

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

내 2 펜스를 던지는 것 같은 느낌.

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

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

이후 파이썬 3.5 당신이 사용할 수있는 os.scandir대신에 os.listdir. 문자열 대신 DirEntry개체 의 반복자를 반환합니다. 문서에서 :

scandir()대신을 사용하면 listdir()파일 유형 또는 파일 속성 정보가 필요한 코드의 성능이 크게 향상 될 수 있습니다 DirEntry. 운영 체제가 디렉터리를 스캔 할 때 제공하는 경우 개체가이 정보를 노출 하기 때문 입니다. 모든 DirEntry방법은 시스템 호출을 수행 할 수 있지만 is_dir()및 is_file()보통 심볼릭 링크에 대한 시스템 호출을 필요로; DirEntry.stat()Unix에서는 항상 시스템 호출이 필요하지만 Windows의 심볼릭 링크에는 하나만 필요합니다.

객체의 이름에 액세스 할 수 있으며 DirEntry.name이를 통해 다음 출력과 동일합니다.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

이것이 내가 그것을 해결 한 방법입니다

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

...

listdir을 사용할 때 캐치가 있습니다. os.path.isdir (식별자)는 절대 경로 여야합니다. 하위 디렉토리를 선택하려면 다음을 수행하십시오.

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

대안은 os.path.join ()없이 테스트를 수행하기 위해 디렉토리로 변경하는 것입니다.

이 스 니펫을 사용할 수 있습니다.

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

제외 목록을 만들고 fnmatch를 사용하여 디렉터리 구조를 건너 뛰고 프로세스를 수행합니다.

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
            ....

'포함'과 동일 :

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

왜 단순히를 사용 range하고 os.walk과 함께 zip? 최상의 솔루션은 아니지만 작동합니다.

예를 들면 다음과 같습니다.

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

파이썬 3에서 나를 위해 작동합니다.

또한 : A break는 너무 간단합니다. (@Pieter의 답변을보세요)

Alex의 대답에 약간의 변경이 있지만 다음을 사용합니다 __next__().

print(next(os.walk('d:/'))[2]) 또는 print(os.walk('d:/').__next__()[2])

와 [2]인 file에 root, dirs, file다른 답변에서 언급

os.walk가 찾은 모든 디렉토리의 루트 폴더 변경. 루트 == 디렉토리인지 확인하는 해결사

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