Đây là một giải pháp với sự hiểu biết danh sách lồng nhau os.walk
và kết hợp hậu tố đơn giản thay vì 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')]
Nó có thể được nén thành một lớp lót:
import os;cfiles=[os.path.join(r,f) for r,d,fs in os.walk('src') for f in fs if f.endswith('.c')]
hoặc khái quát như là một hàm:
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')
Nếu bạn cần glob
mẫu đầy đủ kiểu, bạn có thể làm theo ví dụ của Alex và Bruno và sử dụng 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()
một chút khó sử dụng hơnos.walk()