python读取文件名称并生成txt文件

import os
import numpy as np

def create_txt(name, path, file_image, file_label):
    txt_path = path + name + '.txt'
    txt = open(txt_path, 'w')
    for (i,j) in zip(file_image,file_label):
        image_dir = os.path.join('./dataset/test/', str(i))
        label_dir = os.path.join('./dataset/test_label/', str(j))
        txt.write(image_dir)
        txt.write(label_dir)

def read_file(path1, path2):
    filelist1 = os.listdir(path1)
    file_image = np.array([file for file in filelist1 if file.endswith('.jpg')], dtype=object)
    filelist2 = os.listdir(path2)
    file_label = np.array([file for file in filelist2 if file.endswith('.png')], dtype=object)
    return file_image, file_label


path1 = './dataset/test/'
path2 = './dataset/test_label/'

file_image, file_label = read_file(path1, path2)
create_txt('test', './dataset/', file_image, file_label)