Python判断一个文件夹内哪些文件是图片的实例

Python判断一个文件夹内哪些文件是图片的实例

如下所示:

def is_img(ext):
 ext = ext.lower()
 if ext == '.jpg':
  return True
 elif ext == '.png':
  return True
 elif ext == '.jpeg':
  return True
 elif ext == '.bmp':
  return True
 else:
  return False

调用时

for x in os.listdir(directory):
 if is_img(osp.splitext(x)[1]):
  do something

以上这篇Python判断一个文件夹内哪些文件是图片的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持亿速云。