python 获取文件夹下所有文件列表

off999 2024-09-23 11:26 46 浏览 0 评论

今天上班时候，遇到同事一会啪嗒啪嗒敲键盘，一会抓耳挠腮的，好奇过去问了一下，原来是同事给他一个任务，就是要把一个文件夹内的所有文件包括文件夹路径都整理在表格当中。

一大早就开始一个一个文件的复制文件名，粘贴到表格，然后继续下一个，中途有人跟他说句话，他可能思路随时被打断，不知道自己整理到第几个文件了。

通过文件属性一看，总文件夹下，一共一百多个文件夹，接近一千个文件……

这要整理到啥时候去？

此时python的一个walk浮现在脑海中，来吧，让我试试，半个小时后过来找我要表格吧，多简单的事，直接上代码：

import os

def get_filelist(dir):
    filelist = open('./list.txt', 'w+', encoding='utf8')
    dirlist = open('./dir.txt', 'w+', encoding='utf8')
    d=1
    f=1
    for home, dirs, files in os.walk(dir):
        # 文件夹列表
        for dir in dirs:
            # print(dir)
            dirlist.write(str(d)+'---'+dir)
            dirlist.write('\n')
            d+=1
        # 文件列表
        for filename in files:
            # 拼接全路径
            fullname = os.path.join(home, filename)
            filelist.write(str(f)+'---'+fullname)
            filelist.write('\n')
            f+=1
    dirlist.close()
    filelist.close()

if __name__ == "__main__":
    get_filelist('F:\\测试文件')

运行过程非常快，不到一秒钟，两个文件已经摆在那里了，整整齐齐：

此时的同事已经惊呆了，原计划他是中午不吃饭的，一直干到下班的，现在还不到午饭时间就解决了，这段午饭就有人请吃饭了。

python获取文件名