import numpy as np
import os
import cv2
import csv
import matplotlib as plt
import random
from PIL import Image
import numpy as np
import pandas as pd
test = [] # this is a zhongjianshang
# 对txt文件的操作~
for line in open('new_truth.txt', 'r'):
# print(type(line))
x = line.split('\n')
y = x[0].split(',')
test.append(y)
# 遍历所有的图片
save = []
for i in range(len(test)):
line = []
line.append(test[i][0])
y0 = int(test[i][2])
y1 = int(test[i][4])
x0 = int(test[i][1])
x1 = int(test[i][3])
box1 = [x0, y0, x1, y1]
w = x1 - x0
h = y1 - y0
ares = w * h
line.append(ares)
save.append(line)
title=['name','areas']
x=pd.DataFrame(columns=title,data=save)
x.to_csv('./area.csv')
2、思考
这只是一个样例。
import pandas as pd
step1: 将一行数据保存在一个list中。
step2: 将所有的数据保存在另一个list中。所以变成了[[1,2],[3,4]]。
step3:取个title。
step4:
x=pd.DataFrame(columns=title,data=save)
x.to_csv('./area.csv')