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')



你可能感兴趣的:(Python,Linux)