获取文件夹下文件

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

Python获取指定文件夹下的文件名 - 耸立的青竹 - 博客园

# python
Logo
华为云开发者联盟

为开发者提供学习成长、分享交流、生态实践、资源工具等服务，帮助开发者快速成长。

更多推荐

cover

从原始边列表到邻接矩阵Python实现图数据处理的完整指南

avatar 华为云开发者联盟
cover

华为云云原生FinOps解决方案，释放云原生最大价值

avatar 华为云开发者联盟
cover

解锁HDC 2024之旅：从购票到报名，全程攻略

avatar 华为云开发者联盟