python——獲取文件夾中所有文件的路徑

此腳本功能如下：

輸入一個文件夾路徑，將文件夾中所有文件的路徑存入txt中。

給上代碼：

# -*- coding: utf-8 -*-
# 生成文件夾中所有文件的路徑到txt
import os
def listdir(path, list_name):  # 傳入存儲的list
    for file in os.listdir(path):
        file_path = os.path.join(path, file)
        if os.path.isdir(file_path):
            listdir(file_path, list_name)
        else:
            list_name.append(file_path)

list_name=[]
path='D:/PythonProject/data/'   #文件夾路徑
listdir(path,list_name)
print(list_name)

with open('./list.txt','w') as f:     #要存入的txt
    write=''
    for i in list_name:
        write=write+str(i)+'\n'
    f.write(write)