用python进行文件读写

要求：读取某个文件夹下的所有文件，并将文件名逐行写入txt文件

import os
img_list = os.listdir('./')
with open('img_name.txt', 'w') as f:
    for img in img_list:
        f.write(img+'\n')
# 结果包含所有文件名的txt