python数据写入CSV

python代码中将数据写入CSV表格有两种办法：pandas和CSV。

下面我将介绍什么时候适合使用pandas，什么时候适合使用CSV库。主要区别是一个按行存储方便，一个按列存取方便。

1.按列存数据（使用pandas）

假设第一列为[1,1,1,1],第二列为[2,2,2,2]，第三列的值为[3,3,3]，列名（表头名字）为column1,column2,column3。代码如下：

import pandas as pd
c1= [1,1,1,1]
c2 = [2,2,2,2]
c3= [3,3,3,3]
#use pandas
#write date by using the form of dict
df= pd.DataFrame({'column1':c1,'column2':c2,'column3':c3})
df.to_csv("test1.csv",index=False)

在当前目录生成test1.csv，如图所示：

2.按行存数据（使用CSV）

加入第一行数据为[1,1,1],第二行数据为[2,2,2]，第三行数据为[3,3,3]，第四行数据为[4,4,4]。
代码如下：

import csv
r1=[1,1,1]
r2=[2,2,2]
r3=[3,3,3]
r4=[4,4,4]
#use csv
with open("test2.csv", "w") as csvfile:
    writer = csv.writer(csvfile,lineterminator='\n')
    #first write columns_name
    writer.writerow(["column1","column2","column3"])
    #then write data
    writer.writerows([r1,r2,r3,r4])

在当前目录生成test2.csv，如图所示：

总结：有两种API可以调用，pandas还有CSV，按照不同的需求可以选择不同的处理方式（按行或者按列读取），记得提取导入需要的包。