python实现将子文件夹中的文件地址写入csv

import os
import csv
import pandas as pd

l=[]
files = []
pwd = r'你的文件夹地址'
for filename in os.listdir(pwd):
    l.append(filename) #保存所有子文件夹地址


for line in l:
    for filename in os.listdir(os.path.join(pwd,line)):
        files.append(filename) # 所有文件写入list中
labels = [1 for index in range(len(files))] # csv文件的第二列内容，这里按需改删

with open("test.csv","w",newline='') as f:
    writer = csv.writer(f)
    writer.writerows(zip(files,labels)) # 写入csv文件
f.close()

代码修改：标签设计为图像的类，而非以训练测试集划分，并且增加列名

import os
import csv
import pandas as pd

l=[]
files = []
labels = []
pwd = r'你的文件夹地址'
for filename in os.listdir(pwd):
    l.append(filename)


for index,value in enumerate(l):
    for filename in os.listdir(os.path.join(pwd,value)):
        files.append('images/'+value+'/'+filename)
        labels.append(index)
# labels = [1 for index in range(len(files))]

with open("train.csv","w",newline='',encoding='utf-8') as f:
    writer = csv.writer(f)
    writer.writerow(['img_path','label'])
    writer.writerows(zip(files,labels))
f.close()