使用背景
逛社区发现许多人在解决删除文件夹中非图片文件,删除文件夹中图片等问题的时候,都写了很多代码取实现这一功能,我当时就纳闷了,能几行代码解决的事,为什么要写那么多的代码?
解决下述问题
解决的问题是:删除文件夹中具有相同后缀类型文件。如删除文件夹下所有的JPG文件或者PNG文件
核心函数
str.endswith(suffix[, start[, end]])
函数用法: endswith() 方法用于判断字符串是否以指定后缀结尾,如果以指定后缀结尾返回True,否则返回False。可选参数"start"与"end"为检索字符串的开始与结束位置。
- suffix – 该参数可以是一个字符串或者是一个元素。
- start – 字符串中的开始位置。
- end – 字符中结束位置。
注:返回值为bool类型
实战演练
现在文件夹中有TXT文件(后缀为.txt),CSV文件(后缀为.csv)和PNG文件(后缀为.png),需要删除PNG文件,只保留TXT和CSV文件。代码如下所示:
import os
file=os.listdir(r'C:\Users\ZFG\Desktop\test')
for i in file:
if i.endswith('.png'):
os.remove(r'C:\Users\ZFG\Desktop\test/'+i)
最后展示结果,第一张图为删除前的图片,第二张图为删除后的效果图:
- 注:当文件夹中需要删除的文件很多很多的时候,这几行简单的代码有奇效!
|