获取文件夹下文件
os.listdir() :用于返回指定的文件夹包含的文件或文件夹的名字的列表。得到的是仅当前路径下的文件名,不包括子目录中的文件,所有需要使用递归的方法得到全部文件名。
但是返回的文件名不一定是顺序的,符合需求的,需要重排。
os.walk() :可以遍历文件夹下所有的文件。
os.walk(top, topdown=Ture, onerror=None, followlinks=False)
该函数可以得到一个三元tupple(dirpath, dirnames, filenames
参数含义:
- dirpath:string,代表目录的路径;
- dirnames:list,包含了当前dirpath路径下所有的子目录名字(不包含目录路径);
- filenames:list,包含了当前dirpath路径下所有的非目录子文件的名字(不包含目录路径)。
注意,dirnames和filenames均不包含路径信息,如需完整路径,可使用os.path.join(dirpath, dirnames)
获取扩展名/文件名
使用**os.path.splitext(file)[0]
可获得文件名**。
使用**os.path.splitext(file)[-1]
可获得以.
开头的文件后缀名**。
或者使用 python的 split(‘.’)[0] 和 **split(‘.’)[1]**方法先分割再获取也可。
import os
file = "Hello.py"
# 获取前缀(文件名称)
assert os.path.splitext(file)[0] == "Hello"
# 获取后缀(文件类型)
assert os.path.splitext(file)[-1] == ".py"
assert os.path.splitext(file)[-1][1:] == "py"
示例一
现在需要按照图片的命名顺序逐个读取图片(即按照frame1.jpg,frame2.jpg,frame3.jpg,frame4.jpg,frame5.jpg顺序),通常情况下需要通过文件夹的路径将所有图片名放入到一个列表中,如下imgList保存’E:测试视频01’文件夹下所有图片的名称
import os
imgList = os.listdir('E:测试视频01')
print(imgList)
imgList的内容如下,没有按序排列,
对上一步的imgList进行排序:
import os
imgList = os.listdir('E:测试视频01')
#按照数字进行排序后按顺序读取文件夹下的图片
imgList.sort(key=lambda x: int(x.replace("frame","").split('.')[0]))
print(imgList)
此时 imgList的内容如下,可以看到已经有序排列