获取文件夹下文件
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的内容如下,可以看到已经有序排列
再通过每张图片的文件名获得全路径:
import os
dir = 'E:\测试视频01'
imgList = os.listdir(dir)
print(imgList)
#按照数字进行排序后按顺序读取文件夹下的图片
imgList.sort(key=lambda x: int(x.replace("frame","").split('.')[0]))
print(imgList)
for count in range(0, len(imgList)):
im_name = imgList[count]
im_path = os.path.join(dir,im_name)
print(im_path)
实践:
# frame_list = glob.glob(os.path.join(self.edited_img_dir, path_code, '*g'))
frame_list = os.listdir(os.path.join(self.edited_img_dir, path_code))
try:
frame_list.remove('Thumbs.db')
except:
pass
try:
frame_list.remove('.DS_Store')
except:
pass
print(frame_list)
nums = len(frame_list)
# 按照数字进行排序后按顺序读取文件夹下的图片
frame_list.sort(key=lambda x: int(x.split('.')[0]))
# suffix = os.path.splitext(frame_list[0])[-1]
for i in range(nums):
# img_np = read_img(frame_list[i])
frame_name = os.path.join(self.edited_img_dir, path_code, frame_list[i])
print(frame_name)
img_np = read_img(frame_name)
示例二
from imutils import paths
path_img='output/234/'
# 使用imutils.paths模块中的list_images()方法可以获取当前目录下的图片路径
imagePaths=sorted(list(paths.list_images(path_img)))
print(imagePaths)
# 给读取图像排序遍历,数字由小到大
imagePaths.sort(key=lambda x:int((x.split('/')[-1]).split('.')[0]))
print(imagePaths)
使用imutils.paths模块中的list_images()方法可以获取当前目录下的图片路径。
先使用split('/')进行分割,[-1]表示取出最后一项(即图像名+扩展名);
再用一个split('.')分割文件名,[0]表示取出分割后的第一项(即图像的.前的文件名)。
利用sorted函数对列表字典排序
列表排序
列表排序有两种方法实现,一种是列表自带的sort()方法,即list.sort(),另一种则是通过sorted()进行排序。
sort()方法语法:
list.sort(key=None, reverse=False)
- key -- 主要是用来进行比较的元素,只有一个参数,具体的函数的参数就是取自于可迭代对象中,指定可迭代对象中的一个元素来进行排序。
- reverse -- 排序规则,reverse = True 降序, reverse = False 升序(默认)。
该方法没有返回值,会改变原始列表的顺序。
sorted()语法:
sorted(iterable, key=None, reverse=False)
- iterable -- 可迭代对象。
- key -- 主要是用来进行比较的元素,只有一个参数,具体的函数的参数就是取自于可迭代对象中,指定可迭代对象中的一个元素来进行排序。
- reverse -- 排序规则,reverse = True 降序 , reverse = False 升序(默认)。
返回一个新的列表。
元素是元组或列表
data = [(1, 2), (4, 1), (9, 8), (3, 4)]
# 对元组的第2个元素进行排序
data.sort(key=lambda k: k[1])
print('data:', data)
# 执行结果:data: [(4, 1), (1, 2), (3, 4), (9, 8)]
# 对元组的第1个元素进行排序
print('data:', sorted(data))
# 执行结果:data: [(1, 2), (3, 4), (4, 1), (9, 8)]
元素是字典
data = [{'a': 2}, {'b': 1}, {'c': 9}, {'d': 3}]
# 对字典的值进行排序
data.sort(key=lambda k: list(k.values()))
print('data:', data)
# 执行结果:data: [{'b': 1}, {'a': 2}, {'d': 3}, {'c': 9}]
# 对字典的键进行排序
print('data:', sorted(data, key=lambda k: list(k.keys())))
# 执行结果:data: [{'a': 2}, {'b': 1}, {'c': 9}, {'d': 3}]
元素是嵌套字典
data = [{'a': {'num': 2, 'text': 'a'}}, {'b': {'num': 1, 'text': 'b'}}, {'c': {'num': 9, 'text': 'c'}},
{'d': {'num': 3, 'text': 'd'}}]
# 对嵌套字典中的值进行排序
data.sort(key=lambda k: [i['num'] for i in list(k.values())])
print('data:', data)
# 执行结果:data: [{'b': {'num': 1, 'text': 'b'}}, {'a': {'num': 2, 'text': 'a'}}, {'d': {'num': 3, 'text': 'd'}},
# {'c': {'num': 9, 'text': 'c'}}]
# 对嵌套字典中的值进行排序
print('data:', sorted(data, key=lambda k: [i['text'] for i in list(k.values())]))
# 执行结果:data: [{'a': {'num': 2, 'text': 'a'}}, {'b': {'num': 1, 'text': 'b'}}, {'c': {'num': 9, 'text': 'c'}},
# {'d': {'num': 3, 'text': 'd'}}]
通过自定义函数指定列表中的元素进行排序
data = [{'a': {'num': 2, 'text': 'a'}}, {'b': {'num': 1, 'text': 'b'}}, {'c': {'num': 9, 'text': 'c'}},
{'d': {'num': 3, 'text': 'd'}}]
def key(data):
key_list = []
for i in list(data.values()):
key_list.append(i['num'])
return key_list
# 对嵌套字典中的值进行排序
data.sort(key=key)
print('data:', data)
# 执行结果:data: [{'b': {'num': 1, 'text': 'b'}}, {'a': {'num': 2, 'text': 'a'}}, {'d': {'num': 3, 'text': 'd'}},
# {'c': {'num': 9, 'text': 'c'}}]
# 对嵌套字典中的值进行排序
print('data:', sorted(data, key=key))
# 执行结果:data: [{'b': {'num': 1, 'text': 'b'}}, {'a': {'num': 2, 'text': 'a'}}, {'d': {'num': 3, 'text': 'd'}},
# {'c': {'num': 9, 'text': 'c'}}]
字典排序
因为sorted(data.items())得到的是包含键、值的元组列表,即[('a', 2), ('b', 1), ('c', 9), ('d', 3)],所以可以通过sorted()函数进行排序得到新的列表,再通过字典解析得到字典。
元素是键值对
data = {'a': 2, 'b': 1, 'c': 9, 'd': 3}
# 对值进行排序
data_sorted = {i[0]: i[1] for i in sorted(data.items(), key=lambda k: k[1], reverse=True)}
print('data:', data_sorted)
# 执行结果:data: {'c': 9, 'd': 3, 'a': 2, 'b': 1}
元素是嵌套字典
data = {'a': {'num': 2, 'text': 'a'}, 'b': {'num': 1, 'text': 'b'}, 'c': {'num': 9, 'text': 'c'},
'd': {'num': 3, 'text': 'd'}}
# 对嵌套字典中的值进行排序
data_sorted = {i[0]: i[1] for i in sorted(data.items(), key=lambda k: k[1]['num'])}
print('data:', data_sorted)
# 执行结果:data: {'b': {'num': 1, 'text': 'b'}, 'a': {'num': 2, 'text': 'a'}, 'd': {'num': 3, 'text': 'd'},
# 'c': {'num': 9, 'text': 'c'}}
元素是列表嵌套字典
data = {'a': [{'num1': 2, 'text': 'a'}, {'num2': 3, 'text': 'd'}],
'b': [{'num1': 1, 'text': 'b'}, {'num2': 2, 'text': 'c'}],
'c': [{'num1': 9, 'text': 'c'}, {'num2': 1, 'text': 'b'}],
'd': [{'num1': 3, 'text': 'd'}, {'num2': 9, 'text': 'a'}]}
# 对嵌套字典中的第一个元素进行排序
data_sorted = {i[0]: i[1] for i in sorted(data.items(), key=lambda k: k[1][0]['num1'])}
print('data:', data_sorted)
# 执行结果:data: {'b': [{'num1': 1, 'text': 'b'}, {'num2': 2, 'text': 'c'}],
# 'a': [{'num1': 2, 'text': 'a'}, {'num2': 3, 'text': 'd'}],
# 'd': [{'num1': 3, 'text': 'd'}, {'num2': 9, 'text': 'a'}],
# 'c': [{'num1': 9, 'text': 'c'}, {'num2': 1, 'text': 'b'}]}
参考博客
python按照图片命名顺序读取文件夹下的图片 - 麦克斯的园丁 - 博客园
python: 获取 后缀名(扩展名) / 文件名_JNing-CSDN博客_python 文件名后缀
Python按顺序读取文件夹中文件 - 思念殇千寻 - 博客园
python按照图片命名顺序读取文件夹下的图片_小萝卜头____的博客-CSDN博客_python顺序读取图片
Python3 通过sorted函数实现对列表字典的排序_Klein-的博客-CSDN博客_python3字典排序sorted