利用pandas写入csv文件
第一种方法:直接存入字典数据,key代表列名,键值可以是列表,值的个数要一致
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包,一行一行写入
import csv #python2可以用file替代open
with open("test.csv","w") as csvfile:
writer = csv.writer(csvfile) #先写入columns_name
#设置newline,否则两行之间会空一行
writer.writerow(["index","a_name","b_name"],newline='') #写入多行用writerows
writer.writerows([[0,1,3],[1,2,3],[2,3,4]])
csvfile.close() #文件关闭之后才能看到写入的结果
DataFrame是二维的、大小可变的、成分混合的、具有标签化坐标轴(行和列)的表数据结构。基于行和列标签进行计算。可以被看作是为序列对象(Series)提供的类似字典的一个容器,是pandas中主要的数据结构。
读csv文件:import pandas as pd data = pd.read_csv('test.csv')