
今天教大家如何在一个非常混乱文件夹用程序来寻找所有所需类型的文件,并且输出文件名字
代码如下:
from os import listdir
path = './cat_valid/CAT_06/'
filelist = listdir(path)
filelist1=[]
for i in range(0,len(filelist)):
index = filelist[i].split(".")[-1]
if index=='json':
filelist1.append(filelist[i])
print(filelist1)
首先我们需要导入系统的os库,这个库不需要再下载。
其次我们将需要寻找的文件夹路径赋值给path,在使用listdir()函数将此文件夹下的所有文件转为list类型。
定义filelist1来存放所需文件的名字。
使用for循环遍历此文件夹下所有文件,利用split对其进行分词,获取文件的类型名字。
使用if对所需文件类型进行判断,此处可将‘json’改为所需的‘doc’,‘tex’,‘jpg'等类型,当然也可以多定义几个filelist1列表来对这些文件类型全部统计。
最后运行程序即可输出此文件夹下所有自己所需的文件类型名字。
运行结果为: