当前位置:主页 > 软件编程 > Python代码 >

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

时间:2021-02-22 17:58:24 | 栏目: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

您可能感兴趣的文章:

相关文章