Python读取文件夹中指定格式的文件名与相对路径

近日做人脸数据的提取，需要从文件夹下提取所有图片格式文件的文件名、路径，参考了一篇博客：https://www.cnblogs.com/strongYaYa/p/7200357.html

在其基础上，根据自己的需求进行了改进，可以实现：

1、遍历文件夹，取出jpg格式的图片名称；

2、遍历文件夹，取出jpg格式的图片相对路径。

代码如下：

import os

def get_file_name(path): # 获取目录下所有jpg格式文件的文件名
    image_files = []
    for file in os.listdir(path):
        image_files.append(file)
    image_names = [os.path.splitext(name)[0] for name in image_files if os.path.splitext(name)[1]=='.jpg']
    return image_names

def get_listdir(path):  #获取目录下所有jpg格式文件的地址，返回地址list
    tmp_list = []
    for file in os.listdir(path): 
        if(os.path.splitext(file)[1]=='.jpg'):
            file_path = os.path.join(path, file)  
            tmp_list.append(file_path)
    return tmp_list