使用Python将用户指定的文件夹结构写入CSV文件

Question

我正在尝试编写一个python脚本，它将接受用户指定的目录，然后将该目录的内容写入csv文件。我面临的挑战是将文件夹结构以某种格式写入csv。我需要的格式如下所示：

我试过用手术室步行（dir）列出目录和文件，但我无法以上述格式写入csv。在

我还发现一些代码可以在给定的文件夹结构中创建一个嵌套字典，但是我发现很难在这个嵌套结构中导航并以我需要的方式编写行。在

如果有人能更容易地完成这项任务，我们将不胜感激。在

Answer 1

好的，下面是执行此操作的代码：

import os
import csv

data = []
for root, dirs, files in os.walk('.'):
    for f in files:
        l = root.split('/') + [f]
        data.append(l)

with open('output.csv', 'wb') as f:
    writer = csv.writer(f)
    for d in data:
        writer.writerow(d)

如果您使用的是windows，请将split('/')替换为split('\\')