在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('\\\\')