用python自动将文件名写入txt文件中

import os

img_path = './data./test./p'
img_list = os.listdir(img_path)
f = open('test_p.txt','w')
for img in img_list:
    img = img[:]+'\n'
    f.write(img)
f.close()