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 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เป็นสิ่งที่ดี คำตอบที่ตรงกับคำถามของคุณในหลาม root, dirs, files = os.walk(dir_name).next()2

ไวยากรณ์ Python 3 ที่เทียบเท่าคือ root, dirs, files = next(os.walk(dir_name))

คุณสามารถใช้os.listdir()ซึ่งส่งคืนรายการชื่อ (สำหรับทั้งไฟล์และไดเร็กทอรี) ในไดเร็กทอรีที่กำหนด หากคุณต้องการแยกความแตกต่างระหว่างไฟล์และไดเรกทอรีให้เรียกos.stat()ชื่อแต่ละชื่อ

หากคุณมีข้อกำหนดที่ซับซ้อนมากกว่าไดเร็กทอรีบนสุด (เช่นละเว้น VCS dirs เป็นต้น) คุณยังสามารถแก้ไขรายการไดเร็กทอรีเพื่อป้องกันไม่ให้ 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 ไม่รู้เกี่ยวกับการ rebinding ภายนอก

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

ตั้งแต่Python 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**):

ทำไมไม่ใช้ a 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

ใช้ได้กับฉันใน python 3

นอกจากนี้: A breakนั้นง่ายกว่าด้วย btw (ดูคำตอบจาก @Pieter)

การเปลี่ยนแปลงเล็กน้อยกับคำตอบของ Alex แต่ใช้__next__():

print(next(os.walk('d:/'))[2]) หรือ print(os.walk('d:/').__next__()[2])

กับ[2]การเป็นfileในroot, dirs, fileการกล่าวถึงในคำตอบอื่น ๆ

การเปลี่ยนแปลงโฟลเดอร์รูทสำหรับทุกไดเร็กทอรี os.walk พบ ฉันแก้ปัญหานั้นตรวจสอบว่าไดเรกทอรี 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