ต่อไปนี้เป็นคำตอบของความเข้าใจในรายการแบบซ้อนos.walk
และการจับคู่คำต่อท้ายแบบง่ายแทนglob
:
import os
cfiles = [os.path.join(root, filename)
for root, dirnames, filenames in os.walk('src')
for filename in filenames if filename.endswith('.c')]
มันสามารถบีบอัดให้เป็นหนึ่งซับ:
import os;cfiles=[os.path.join(r,f) for r,d,fs in os.walk('src') for f in fs if f.endswith('.c')]
หรือทำเป็นฟังก์ชั่นทั่วไป:
import os
def recursive_glob(rootdir='.', suffix=''):
return [os.path.join(looproot, filename)
for looproot, _, filenames in os.walk(rootdir)
for filename in filenames if filename.endswith(suffix)]
cfiles = recursive_glob('src', '.c')
หากคุณต้องการรูปglob
แบบเต็มคุณสามารถทำตามตัวอย่างของ Alex และ Bruno และใช้fnmatch
:
import fnmatch
import os
def recursive_glob(rootdir='.', pattern='*'):
return [os.path.join(looproot, filename)
for looproot, _, filenames in os.walk(rootdir)
for filename in filenames
if fnmatch.fnmatch(filename, pattern)]
cfiles = recursive_glob('src', '*.c')
os.path.walk()
มีข้อผิดพลาดเล็กน้อยที่จะใช้มากกว่าos.walk()