使用Python搜索JPEG文件

Question

我想用Python脚本在一个文件夹里搜索jpeg格式的图片文件，并列出这些文件的名字。有没有人能帮我一下，教我怎么识别jpeg图片文件。

提前谢谢大家...

Answer 1

使用 magic 模块来获取文件的MIME类型，然后查找 image/jpeg。

Answer 2

如果你想扫描子文件夹：

import os

for root, subdirs, files in os.walk(DIRECTORY):
    for file in files:
        if os.path.splitext(file)[1].lower() in ('.jpg', '.jpeg'):
             print os.path.join(root, file)

否则，你可以使用其他答案中的一些 glob 函数，或者这个：

import os

for f in os.listdir(DIRECTORY):
    if os.path.splitext(f)[1].lower() in ('.jpg', '.jpeg'):
        print os.path.join(DIRECTORY, f)

应该也能正常工作。

Answer 3

如果你想在一个文件夹里搜索文件，而且不想搜索子文件夹里的内容，你可以直接这样做：

>>> import glob
>>> glob.glob("D:\\bluetooth\*.jpg")
['D:\\bluetooth\\Image1475.jpg',  'D:\\bluetooth\\Image1514.jpg']

想了解更多关于 glob 的信息，可以点击这里。这个工具可以让你像在Unix系统里一样使用通配符进行搜索，比如：

>>> import glob
>>> glob.glob('./[0-9].*')
['./1.gif', './2.txt']
>>> glob.glob('*.gif')
['1.gif', 'card.gif']
>>> glob.glob('?.gif')
['1.gif']