保存文件为csv

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')