在Python

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