python寫入csv文件的幾種方法總結

 

生成test.csv文件

#coding=utf-8
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

 

讀取csv文件

#coding=utf-8
import pandas as pd
data = pd.read_csv('test.csv')
print(data)

輸出

C:\Users\macname\Desktop\t>python test.py
   a_name  b_name
0       1       4
1       2       5
2       3       6

 

---

另一種方法用csv包，一行一行寫入

#coding=utf-8
import csv

#python2可以用file替代open
with open("test.csv","w") as csvfile: 
    writer = csv.writer(csvfile)

    #先寫入columns_name
    #寫入多行用writerows
    writer.writerows([["index","a_name","b_name"],[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

#coding=utf-8
import csv
with open("test.csv","r") as csvfile:
    reader = csv.reader(csvfile)
    #這里不需要readlines
    for line in reader:
        print(line)

輸出

C:\Users\macname\Desktop\t>python test.py
['index', 'a_name', 'b_name']
[]
['0', '1', '3']
[]
['1', '2', '3']
[]
['2', '3', '4']
[]