python 怎么获取本地图片的路径

Question

指定一个文件夹img，img有俩个子文件夹a和b，图片格式有png，jpg，获取a文件夹里面图片的路径存到列表al，获取b文件夹图片的路径存到列表bl，怎么做到，要代码

Answer 1

小试了一下

#! /usr/bin/env python3
# _*_ coding: utf8 _*_

import os

al = []
bl = []


def img_add_path(dir, subdir=''):
    """
    获取绝对路径分类
    :param dir: 父目录
    :param subdir: 子目录
    :return: 无
    """
    for file in os.listdir(f'{dir}/{subdir}'):
        local_path = f'{dir}/{subdir}/{file}'
        if local_path.endswith('.png'):
            al.append(local_path)
        elif local_path.endswith('.jpg'):
            bl.append(local_path)
        else:
            print(local_path)


if __name__ == '__main__':
    path_img = '/home/img'
    # a 目录
    img_add_path(path_img, 'a')
    # b 目录
    img_add_path(path_img, 'b')