python读,写csv文件的2种方法

最常用的一种方法,利用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
 

猜你喜欢

转载自blog.csdn.net/fu6543210/article/details/83240127
今日推荐