如何从目录中打开文件？

Question

import Image
sizes = [(120,120), (720,720), (1600,1600)]
files = ['a.jpg','b.jpg','c.jpg']

for image in files:
    for size in sizes:
      Image.open(image).thumbnail(size).save("thumbnail_%s" % image)

如何从目录中打开文件？因此输入将是目录，而不是files。

Answer 1

您可以使用glob：

from glob import glob
files = glob('%s/*.jpg' % mydir)
for image in files:
    (...)

Answer 2

您可以使用os.listdir获取目录中的文件列表。您可以使用os.path.isfile过滤文件，使用re.match查找与特定名称/扩展名匹配的文件。

这样的事情：

import re
import os

folder = 'images'
files = [f for f in os.listdir(folder) if os.path.isfile(os.path.join(folder, f))]
jpegs = [f for f in files if re.match(r'.*\.jpe?g', f) != None]