python之获取目录下的文件名并写入文件

背景：为了统计一个目录下所有文件名，并且写到文件中

工具：python的模块os.walk

功能：遍历目录下的所有子文件及其子文件夹，如果有子文件夹的话，会继续以子文件夹为目录递归遍历

返回：如果只有文件的话，只返回一个元组（当前目录字符串，文件夹名列表，文件名列表）；如果还有文件夹的话，返回多个元组

code demo:

#!/usr/bin/python
#coding=utf-8

import os
mydir = r'/home/test/'

if __name__ == '__main__':
    for pathstr,dirlist,filelist in os.walk(mydir):
        for file in filelist:
            with open("file.txt","ab") as f:
                f.write(file+'\n')