python读写csv文件并增加行列的实例代码
python读写csv文件并增加行列，具体代码如下所示：
#-*-coding:utf-8-*-
"""
CreatedonThuAug1711:28:172017
@author:ShawnYuen
"""
importcsv
d=list(range(38685))
withopen('./kinetics_test.csv')asf1:
f_csv=csv.DictReader(f1)
fori,rowinenumerate(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']
withopen('./kinetics_test_new_.csv','w')asf:
f_csv=csv.DictWriter(f,headers)
f_csv.writeheader()
f_csv.writerows(d)
f.close()
withopen('./kinetics_test_new_.csv','rt')asfin:
lines=''
forlineinfin:
ifline!='\n':
lines+=line
withopen('./kinetics_test_new.csv','wt')asfout:
fout.write(lines)
为了下载test数据，
方法一:用Excel打开csv文件，手动添加label和is_cc;
方法二:利用python里面的csv模块改写。
生成的csv文件中奇怪的多了一些空行，然后找到解决方法，见参考资料。
python使用writerows写csv文件产生多余空行
python定义给定初值或长度的list
知识点扩展：
python写入csv文件的几种方法总结
最常用的一种方法，利用pandas包
importpandasaspd
#任意的多组列表
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_nameb_name
014
125
236
同样pandas也提供简单的读csv方法
importpandasaspd
data=pd.read_csv('test.csv')
会得到一个DataFrame类型的data，不熟悉处理方法可以参考pandas十分钟入门
另一种方法用csv包，一行一行写入
importcsv
#python2可以用file替代open
withopen("test.csv","w")ascsvfile:
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]])
indexa_nameb_name
013
123
234
读取csv文件用reader
importcsv
withopen("test.csv","r")ascsvfile:
reader=csv.reader(csvfile)
#这里不需要readlines
forlineinreader:
printline
总结
以上所述是小编给大家介绍的python读写csv文件并增加行列的实例代码,希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对毛票票网站的支持！
如果你觉得本文对你有帮助，欢迎转载，烦请注明出处，谢谢！