python處理csv類型文件的方法

小編給大家分享一下python處理csv類型文件的方法，希望大家閱讀完這篇文章后大所收獲，下面讓我們一起去探討吧！

下面講一下常見的幾種辦法處理csv文件：

最常用的一種方法，利用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存儲(chǔ)為csv,index表示是否顯示行名，default=True
dataframe.to_csv("test.csv",index=False,sep=',')

輸出結(jié)果

  a_name  b_name
0       1       4
1       2       5
2       3       6

同樣pandas也提供簡(jiǎn)單的讀csv方法

import pandas as pd
data = pd.read_csv('test.csv')

另一種方法用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]])

輸出結(jié)果

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

看完了這篇文章，相信你對(duì)python處理csv類型文件的方法有了一定的了解，想了解更多相關(guān)知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道，感謝各位的閱讀！