How to delete files with the same suffix type in a folder in Python

Use background

Walking around the community, I found that many people wrote a lot of code to implement this function when solving problems such as deleting non-picture files in folders and deleting pictures in folders. I then I'm confused, why do I need to write so much code when something can be solved with a few lines of code?

Solve the following problems

The problem solved is: delete files with the same suffix type in the folder. For example, delete all JPG files or PNG files in the folder

Core function

str.endswith(suffix[, start[, end]])

Function usage: The endswith() method is used to determine whether the string ends with the specified suffix. If it ends with the specified suffix, it returns True, otherwise it returns False. The optional parameters "start" and "end" are the start and end positions of the retrieved string.

• suffix – This parameter can be a string or an element.

• start – The starting position in the string.

• end – Ending position in characters.

Note: The return value is bool type

Practical exercise

Now there are TXT files (suffix .txt) and CSV files ( csv) and PNG files (suffix .png), you need to delete the PNG files and keep only the TXT and CSV files. The code is as follows:

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保存的只是文件夹中的一个相对路径，绝对路径还需要加上文件夹的路径+‘/'

The final result is displayed. The first picture is the picture before deletion, and the second picture is the effect picture after deletion:

Note: When there are many files that need to be deleted in the folder, these simple lines of code can work wonders!

Attachment: The following code deletes jpg images in the specified directory

# 删除文件夹下面的图片

#先导入所需的包
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)

The above is the detailed content of How to delete files with the same suffix type in a folder in Python. For more information, please follow other related articles on the PHP Chinese website!