最常用的一种方法,利用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十分钟入门
=======================================上面方法你得安装pandas包,懒人模式直接用下面的方法。
另一种方法用csv包,一行一行写入
import csv
#没有该文件会自动创建,也可以指定存到文件夹里面,记得文件夹路径别放到其他目录下,否则此bug真的难找啊。
# w形式的,下次继续写的话,重头开始,会覆盖掉原来的内容。若要多次写入同一文件内,应改为 a模式,尾部追加内容。
with open("test.csv","w") as csvfile:
writer = csv.writer(csvfile)
#先写入columns_name
writer.writerow(["index","a_name","b_name"])
#写入多行用writerows,list里面每个元素写一行。
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
---------------------
作者:waple_0820
原文:https://blog.csdn.net/waple_0820/article/details/70049953