僅限os.walk .jpg

Question

我只想要以.jpg或.pdf結尾的文件名。我如何限制文件名搜索？

import os 
from subprocess import call 

for dirname, dirnames, filenames in os.walk('.'): 
    for filename in filenames: 
     jpg = os.path.join(dirname, filename) 
     call(["./curl_recognize.sh", jpg, jpg+".txt", "-f txt"]) 

Answer 1

如何：

for filename in filenames: 
    if not filename.endswith('.jpg'): 
    continue 

Answer 2

如果你想在結束所有文件的列表，你可以做這樣的事情

filelist = [] 
for root, dirs, files in os.walk('.'): 
    filelist = filelist + [os.path.join(x,root) for x in files if x.endswith(('.jpg','.pdf'))] 

Answer 3

import fnmatch 
import os 

for file in os.listdir('.'): 
    if fnmatch.fnmatch(file, '*.txt'): 
     print file 

的fnmatch優於其他方法，因爲它假設unix像通配符，包括：

* =匹配一切

？=匹配任何單個字符

[SEQ] =匹配序列的任何字符

[！SEQ] =不SEQ