python判断文件后缀,python判断文件夹内是否存在指定后缀文件

该代码主要是基于python实现判断指定文件夹下是否存在指定后缀的文件。代码如下：

import os

Your_Dir='你的文件夹/'

Files=os.listdir(Your_Dir)

for k in range(len(Files)):

# 提取文件夹内所有文件的后缀

Files[k]=os.path.splitext(Files[k])[1]

Str2=['.wav','.mp3','.mp4']

if len(list(set(Str2).intersection(set(Files))))==len(Str2):

return True

else:

return False

上面代码可以判断是否有指定的一种后缀或者多种后缀同时有。如果只判断是否存在某种指定的一个后缀的话，也可以直接：

import os

Your_Dir='D:/python资料/01linux基础视频/课程视频/'

Files=os.listdir(Your_Dir)

for k in range(len(Files)):

# 提取文件夹内所有文件的后缀

Files[k]=os.path.splitext(Files[k])[1]

# 你想要找的文件的后缀

Str='.mp4'

if Str in Files:

return True

else:

return False

代码中的 os.path.splitext() 函数主要是分离文件的文件名和文件后缀。如一个文件的文件名为:

file='123.wav'

则os.path.splitext(file)[0] = '123' ; os.path.splitext(file)[1] ='.wav'

若错误地方请指出。