python如何顺序读取文件夹下的图片及文件

获取文件夹下文件

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获取指定文件夹下的文件名 – 耸立的青竹 – 博客园

    来源:m0_61899108

    物联沃分享整理
    物联沃-IOTWORD物联网 » python如何顺序读取文件夹下的图片及文件

    发表评论