在 Python 中将用户指定的文件夹结构写入 csv

Question

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

我尝试使用 os.walk(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('\\')