python写csv文件的方法

2023-06-12,

这篇文章给大家分享的是有关pythoncsv文件的方法的内容。小编觉得挺实用的,因此分享给大家做个参考。一起跟随小编过来看看吧。

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

另一种方法用csv包,一行一行写入。

import csv
 
#python2可以用file替代open
with open("test.csv","w") as csvfile: 
    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]])
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

感谢各位的阅读!关于python写csv文件的方法就分享到这里了,希望以上内容可以对大家有一定的帮助,让大家可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到吧!