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