python遍历指定文件夹下面的文件名字

在我们做深度学习的数据处理的时候，需要用到数据集的 train.txt ，也就是遍历指定路径文件夹下面的图片路径和图片名字，然后生成一个 train.txt ，下面是python的一种遍历方法。

import os
import numpy as np
def create_txt(name, path, file_image):
    txt_path = path + name + '.txt'
    txt = open(txt_path, 'w')
    for i in (file_image):
        print(i)
        image_dir = os.path.join('C:/Users/lenovo/Desktop/打标分类/sum/sample/',str(i)+'\n')
        txt.write(image_dir)
def read_file(path1):
    filelist1 = os.listdir(path1)
    file_image = [file for file in filelist1 if file.endswith('.jpg')]
    return file_image
path1 = 'C:/Users/lenovo/Desktop/打标分类/sum/sample/'
file_image = read_file(path1)
create_txt('train', '', file_image)

运行后生成的 train.txt 会在这个 py 文件的同级目录下。