python中如何写入csv文件
第一种方法:利用pandas:直接建DataFrame对象存入字典数据
import pandas as pd #任意的多组列表
a = [1,2,3] b = [4,5,6] #字典中的key值即为csv中列名 ,键值可以是列表,值的个数要一致
df = pd.DataFrame({'a_name':a,'b_name':b}) #将DataFrame存储为csv, index表示是否显示行名,default=True
df.to_csv("test.csv",index=False,sep=',')
#读csv文件:
data = pd.read_csv('test.csv')
第二种方法:利用csv包,一行一行写入,要求先打开这个文件
import csv #python2可以用file替代open
with open("test.csv","w",newline='') as csvfile: #设置newline,否则两行之间会空一行
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]])
csvfile.close() #文件关闭之后才能看到写入的结果
#用reder读取csv文件
with open('test.csv','r') as csvFile:
reader = csv.reader(csvFile)
for line in reader:
print line
DataFrame是二维的、大小可变的、成分混合的、具有标签化坐标轴(行和列)的表数据结构。基于行和列标签进行计算。可以被看作是为序列对象(Series)提供的类似字典的一个容器,是pandas中主要的数据结构。