如何修改Python脚本实现CSV文件同时写入文件路径与Label列?
修改后的脚本
直接在searchResults列表中同时添加文件完整路径和Label值,就能让CSV每行写入两列内容,修改后的完整代码如下:
import csv import os searchDirPath = '/mnt' searchFileExt = '.mp4' with open('/root/val_25.csv', 'w+', newline='') as outfile: writer = csv.writer(outfile, lineterminator='\n') for root, dirs, files in os.walk(searchDirPath): for file in files: searchResults = [] if file.endswith(".mp4"): full_address = os.path.normpath(os.path.join(root, file)) path_components = full_address.split(os.sep) Label = path_components[6] # label name # 同时添加文件路径和Label到列表 searchResults.append(str(full_address)) searchResults.append(Label) print(os.path.join(root, file), Label) writer.writerow(searchResults)
关键改动说明
- 在
searchResults.append(str(full_address))之后,新增一行searchResults.append(Label),让列表包含两个元素,对应CSV的两列 - 可选:修改print语句,同时打印路径和Label,方便验证输出是否正确
另外注意:path_components[6]依赖固定的目录层级结构,如果后续目录层级调整,这个索引需要同步修改,避免出现索引越界错误。
内容的提问来源于stack exchange,提问作者dtr43
相关产品推荐
相关产品推荐

