您的位置：首页 > 编程语言 > Python开发

python读写csv文件并增加行列的实例代码

2019-08-01 10:57 1076 查看

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写入csv文件的几种方法总结

最常用的一种方法，利用pandas包

import pandas as pd
#任意的多组列表
a = [1,2,3]
b = [4,5,6]
#字典中的key值即为csv中列名
dataframe = pd.DataFrame({'a_name':a,'b_name':b})
#将DataFrame存储为csv,index表示是否显示行名，default=True
dataframe.to_csv("test.csv",index=False,sep=',')
a_name b_name
0 1 4
1 2 5
2 3 6

同样pandas也提供简单的读csv方法

import pandas as pd
data = pd.read_csv('test.csv')

会得到一个DataFrame类型的data，不熟悉处理方法可以参考pandas十分钟入门

另一种方法用csv包，一行一行写入

import csv
#python2可以用file替代open
with open("test.csv","w") as csvfile:
writer = csv.writer(csvfile)
#先写入columns_name
writer.writerow(["index","a_name","b_name"])
#写入多行用writerows
writer.writerows([[0,1,3],[1,2,3],[2,3,4]])

index a_name b_name
0 1 3
1 2 3
2 3 4

读取csv文件用reader

import csv
with open("test.csv","r") as csvfile:
reader = csv.reader(csvfile)
#这里不需要readlines
for line in reader:
print line

总结

以上所述是小编给大家介绍的python读写csv文件并增加行列的实例代码,希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对脚本之家网站的支持！
如果你觉得本文对你有帮助，欢迎转载，烦请注明出处，谢谢！

您可能感兴趣的文章:

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签： python csv 文件

相关文章推荐

新的分享

章节导航