python删除文件夹中具有相同后缀类型文件

使用背景

逛社区发现许多人在解决删除文件夹中非图片文件,删除文件夹中图片等问题的时候,都写了很多代码取实现这一功能,我当时就纳闷了,能几行代码解决的事,为什么要写那么多的代码?

解决下述问题

解决的问题是:删除文件夹中具有相同后缀类型文件。如删除文件夹下所有的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)#i保存的只是文件夹中的一个相对路径,绝对路径还需要加上文件夹的路径+‘/’

最后展示结果,第一张图为删除前的图片,第二张图为删除后的效果图:

  • 注:当文件夹中需要删除的文件很多很多的时候,这几行简单的代码有奇效!

猜你喜欢

转载自blog.csdn.net/weixin_50233398/article/details/120189866