【python】批量处理csv和txt文件(csv和txt相互转换+csv的读取和写入+txt的读取和写入)

  1. 读取csv文件中的信息,将该信息涉及到的图片,批量保存到另一个目录下:
import cv2
import os
import csv

csvfile = open("/*/models-master/research/slim/mobilenet_v2_place6/placerightresult.csv","r")
reader = csv.reader(csvfile)

index = 0
for item in reader:
    print(item[0])
    filename = item[0]
    img = cv2.imread(filename)
    for i in range(len(filename)-1,-1,-1):
        if filename[i] == '/':
            index = i
            break
    print(filename[(index + 1):])
    cv2.imwrite('/*/models-master/research/slim/mobilenet_v2_place6/code/testdataset'+"/"+filename[(index+1):],img) 
  1. 根据一个csv文件,重写另一个csv文件(对其中的部分变量名进行更改工作):
import cv2
import os
import csv

csvfile = open("/*/models-master/research/slim/mobilenet_v2_place6/placerightresult.csv","r")
reader = csv.reader(csvfile)
newcsv = open("/*/models-master/research/slim/mobilenet_v2_place6/code/result700.csv",'a+',newline='')

index = 0
for item in reader:
    print(item[0])
    filename = item[0]
    img = cv2.imread(filename)
    for i in range(len(filename)-1,-1,-1):
        if filename[i] == '/':
            index = i
            break
    print(filename[(index + 1):])
    writer = csv.writer(newcsv)
    row=[filename[(index + 1):-4], item[1]]
    writer.writerow(row)
  1. 利用excel可以直接对csv文件进行排序(升序、降序均可),对于排序后的csv文件,将其转换成txt文件:
import cv2
import os
import csv
newcsv = open("/*/research/slim/mobilenet_v2_place6/code/result700.csv","r")
resulttxt = "/*/research/slim/mobilenet_v2_place6/code/result700.txt"

reader = csv.reader(newcsv)


with open(resulttxt,'a+', encoding='utf-8') as f:
    for item in reader:
        f.write(item[0]+'.jpg'+' '+item[1]+'\n')

猜你喜欢

转载自blog.csdn.net/qq_43348528/article/details/108796264