ไม่รวมไดเรกทอรีใน os.walk

ฉันกำลังเขียนสคริปต์ที่ลงไปในแผนผังไดเรกทอรี (โดยใช้ os.walk ()) จากนั้นเยี่ยมชมแต่ละไฟล์ที่ตรงกับนามสกุลไฟล์ที่แน่นอน อย่างไรก็ตามเนื่องจากต้นไม้ไดเรกทอรีบางส่วนที่เครื่องมือของฉันจะใช้ในนั้นยังมีไดเรกทอรีย่อยที่ในทางกลับกันมีจำนวนมากของไร้ประโยชน์ (สำหรับวัตถุประสงค์ของการสคริปต์นี้) สิ่งที่ฉันคิดว่าฉันต้องการเพิ่มตัวเลือกสำหรับผู้ใช้เพื่อระบุ รายการไดเรกทอรีที่จะแยกออกจากการแวะผ่าน

มันง่ายพอกับ os.walk () ท้ายที่สุดมันก็ขึ้นอยู่กับฉันที่จะตัดสินใจว่าจริง ๆ แล้วฉันต้องการไปที่ไฟล์ / dirs ที่เกี่ยวข้องโดย os.walk () หรือข้ามพวกเขา ปัญหาคือถ้าฉันมีต้นไม้ไดเรกทอรีเช่นนี้

root--
     |
     --- dirA
     |
     --- dirB
     |
     --- uselessStuff --
                       |
                       --- moreJunk
                       |
                       --- yetMoreJunk

และฉันต้องการยกเว้นuselessStuffและลูก ๆ ของมัน os.walk () จะยังคงสืบเชื้อสายมาจากไดเรกทอรีย่อย (อาจมีหลายพันรายการ) ของuselessStuffซึ่งไม่จำเป็นต้องพูดช้าลงสิ่งต่าง ๆ มากมาย ในโลกอุดมคติฉันสามารถบอก os.walk () เพื่อไม่ให้รำคาญที่จะให้ลูก ๆ ที่ไร้ประโยชน์อีกต่อไป แต่เพื่อความรู้ของฉันไม่มีทางที่จะทำเช่นนั้นได้ (มีไหม)

ไม่มีใครมีความคิด? อาจมีห้องสมุดบุคคลที่สามที่ให้สิ่งนั้นใช่ไหม

การแก้ไขแบบdirs in-placeจะตัดไฟล์และไดเร็กทอรีที่ตามมาที่เข้าชมโดยos.walk:

# exclude = set([...])
for root, dirs, files in os.walk(top, topdown=True):
    dirs[:] = [d for d in dirs if d not in exclude]

จากความช่วยเหลือ (os.walk):

เมื่อ topdown เป็นจริงผู้เรียกสามารถแก้ไขรายการ dirnames แบบแทน (เช่นผ่านการมอบหมายเดลหรือสไลซ์) และการเดินจะลดค่าลงในไดเรกทอรีย่อยที่ชื่อยังคงอยู่ใน dirnames; สามารถใช้เพื่อตัดการค้นหา ...

... รูปแบบทางเลือกของคำตอบที่ยอดเยี่ยมของ @ unutbu ที่อ่านโดยตรงอีกเล็กน้อยเนื่องจากความตั้งใจที่จะแยกไดเรกทอรีในราคา O (n ** 2) เทียบกับเวลา O (n)

(การทำสำเนาของรายการ dirs ด้วยlist(dirs)จำเป็นสำหรับการดำเนินการที่ถูกต้อง)

# exclude = set([...])
for root, dirs, files in os.walk(top, topdown=True):
    [dirs.remove(d) for d in list(dirs) if d in exclude]