读取csv文件计数并转存为csv文件 (备忘待优化)

读取csv文件计数并转存为csv文件

读取已划分好的数据集，并统计数据集中不同表情标签的个数，最后存为csv文件。

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import numpy as np
import cv2 as cv
import pandas as pd
import csv


# print(a)
list = []
for i in range(5):
    sf = "D:\zhang\zhangT/valence/sd" + str(i) + ".csv"
    files = csv.DictReader(open(sf,"r"))
    # print(files)
    # cv.waitKey(100000)
    a = [0] * 19
    for file in files:
        L = file['valence']
        label = int(float(L))
        if label == 9: a[0] += 1
        if label == 8: a[1] += 1
        if label == 7: a[2] += 1
        if label == 6: a[3] += 1
        if label == 5: a[4] += 1
        if label == 4: a[5] += 1
        if label == 3: a[6] += 1
        if label == 2: a[7] += 1