修改前:filename一列文件名无后缀
创建modify_csv.py文件,内容如下:
# -*- coding:utf-8 -*-
import numpy as np
import pandas as pd
train_filename = pd.read_csv("train_labels.csv") # csv文件
# 对csv文件中的filename一列的数据加上后缀.jpg
train_filename['filename'] = train_filename['filename'].apply(lambda x:str(int(x))+'.jpg')
train_filename.to_csv("train_labels_new.csv", index=None)
test_filename = pd.read_csv("test_labels.csv")
test_filename['filename'] = test_filename['filename'].apply(lambda x:str(int(x))+'.jpg')
test_filename.to_csv("test_labels_new.csv", index=None)
修改后:filename一列文件名批量加上后缀.jpg