python实现将一个文件夹下的图像文件路径写入到指定的txt文件中

import os

def file_write_txt(path,txt_file):
   f = open(txt_file,'w')
   filenames=os.listdir(path)
   for filename in filenames:
    if os.path.splitext(filename)[1]=='.txt':  # 写入的文件后缀
        out_path=path+filename
        print(out_path)
        # f.writable(out_path+'\n')
        f.write(out_path+'\n')
   f.close()

path='./test2/' # 文件路径
txt_file = path + '../train.txt'
file_write_txt(path,txt_file)

./test2/RK06BB8U053117_01_142344_ok.txt
./test2/RK03AA8U143327_01_144137_ok.txt
./test2/RK03AA8U143023_02_142836_ok.txt
./test2/RK06BB9U212359_02_145243_ok.txt
./test2/RK03AA8U143327_03_144147_ok.txt
./test2/RK06BB8U053211_01_142413_ok.txt
./test2/RK05AA3U101959_03_143601_ok.txt
./test2/RK06AA9U084226_03_150016_ok.txt
./test2/RK06BB8U053820_01_142707_ok.txt
./test2/RK03AA8U143327_02_144142_ok.txt
./test2/RK06BB9U212606_01_143432_ok.txt
./test2/RK06BB9U212359_01_145237_ok.txt
./test2/RK03AA8U143207_03_145802_ok.txt
./test2/RK06BB8U053728_02_142255_ok.txt
./test2/RK06BB8U053728_03_142301_ok.txt
./test2/RK05AA3U101649_03_143415_ok.txt
./test2/RK06BB9U212606_02_143446_ok.txt
./test2/RK06AA9U085550_03_151321_ok.txt
./test2/RK03AA8U143023_03_142841_ok.txt
./test2/RK06BB8U053240_01_142110_ok.txt
./test2/RK06AA9U085629_01_150921_ok.txt
./test2/RK06BB8U053728_01_142243_ok.txt
./test2/RK06BB9U212359_03_145247_ok.txt
./test2/RK06AA8U082943_01_144046_ok.txt
./test2/RK06BB8U053240_02_142119_ok.txt
./test2/RK06AA9U085550_02_151316_ok.txt
./test2/RK06BB8U053240_03_142125_ok.txt
./test2/RK03AA8U142922_02_145444_ok.txt
./test2/RK06AA9U085629_03_150931_ok.txt
./test2/RK03AA8U143207_01_145749_ok.txt
./test2/RK06BB9U212719_02_143527_ok.txt
./test2/RK06BB8U053820_02_142713_ok.txt
./test2/RK06AA8U092236_02_150300_ok.txt
./test2/RK05AB9U223718_02_145510_ok.txt
./test2/RK06AA9U085629_02_150926_ok.txt
./test2/RK05AA3U101959_01_143548_ok.txt
./test2/RK05AA3U101255_02_143225_ok.txt
./test2/RK06AA8U092236_01_150253_ok.txt
./test2/RK03AA8U142957_02_142739_ok.txt
./test2/RK06AA8U081637_03_144718_ok.txt
./test2/RK05AB9U223718_01_145504_ok.txt
./test2/RK03AA8U142922_01_145435_ok.txt
./test2/RK03AA8U142957_03_142744_ok.txt
./test2/RK06BB9U212606_03_143453_ok.txt
./test2/RK05AB9U224051_03_145703_ok.txt
./test2/RK06AA9U090050_01_150810_ok.txt
./test2/RK06BB8U053211_03_142425_ok.txt
./test2/RK06AA9U084226_02_150011_ok.txt
./test2/RK05AA3U101255_01_143206_ok.txt
./test2/RK05AB9U223718_03_145525_ok.txt
./test2/RK06BB9U212719_03_143532_ok.txt
./test2/RK06AA8U082943_03_144120_ok.txt
./test2/RK06BB9U212719_01_143509_ok.txt
./test2/RK06AA8U081637_01_144708_ok.txt
./test2/RK03AA8U142957_01_142733_ok.txt
./test2/RK03AA8U142922_03_145448_ok.txt
./test2/RK05AB9U224051_02_145658_ok.txt
./test2/RK06BB8U053117_03_142357_ok.txt
./test2/RK06AA8U082943_02_144114_ok.txt
./test2/RK05AA3U101649_01_143342_ok.txt
./test2/RK03AA8U143023_01_142832_ok.txt
./test2/RK06BB8U053117_02_142352_ok.txt
./test2/RK05AB9U224051_01_145647_ok.txt
./test2/RK06AA8U081637_02_144713_ok.txt
./test2/RK06AA9U090050_03_150821_ok.txt
./test2/RK05AA3U101255_03_143230_ok.txt
./test2/RK06AA9U090050_02_150815_ok.txt
./test2/RK06AA9U084226_01_150005_ok.txt
./test2/RK05AA3U101959_02_143555_ok.txt
./test2/RK06AA8U092236_03_150306_ok.txt
./test2/RK06AA9U085550_01_151301_ok.txt
./test2/RK06BB8U053211_02_142419_ok.txt
./test2/RK06BB8U053820_03_142717_ok.txt
./test2/RK03AA8U143207_02_145755_ok.txt
./test2/RK05AA3U101649_02_143411_ok.txt