You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何修改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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.22 04:59:58