「Python」python读写csv文件并增加行列

# -*- coding: utf-8 -*-
"""
Created on Thu Aug 17 11:28:17 2017

@author: Shawn Yuen
"""

import csv
d = list(range(38685))
with open('./kinetics_test.csv') as f1:
    f_csv = csv.DictReader(f1)
    for i, row in enumerate(f_csv):
        #print(row)
        key1 = 'label'
        value1 = 'test'
        row[key1] = value1
        key2 = 'is_cc'
        value2 = '0'
        row[key2] = value2
        d[i] = row
f1.close()
headers = ['label', 'youtube_id', 'time_start', 'time_end', 'split', 'is_cc']
with open('./kinetics_test_new_.csv', 'w') as f:
    f_csv = csv.DictWriter(f, headers)
    f_csv.writeheader()
    f_csv.writerows(d)
f.close()
with open('./kinetics_test_new_.csv','rt') as fin:
    lines=''
    for line in fin:
        if line!='\n':
            lines+=line
with open('./kinetics_test_new.csv','wt')as fout:
    fout.write(lines)

为了下载test数据,方法一: 用Excel打开csv文件,手动添加label和is_cc; 方法二: 利用python里面的csv模块改写。生成的csv文件中奇怪的多了一些空行,然后找到解决方法,见参考资料。

python使用writerows写csv文件产生多余空行

python 定义给定初值或长度的list

你可能感兴趣的:(「Python」python读写csv文件并增加行列)