删除制定目录下的指定后缀名的文件

本代码是删除指定目录下的jpg图片

# 删除文件夹下面的图片

#先导入所需的包
import os
#指定路径
path = '/content/drive/My Drive'
filenames = os.listdir(path)
for filename in filenames:
    print(filename)
    ＃判断是否是文件而非文件夹
    if(os.path.isfile(filename)):
#         print('it is a file')
		＃判断文件后缀名是否是‘.jpg’
        if(os.path.splitext(filename)[1] == '.jpg'):
        　#执行删除
            os.remove(filename)