python如何读取csv数据

python读取csv数据的方法：首先利用csv.reader方法来读取csv文件，该方法会返回一个可迭代的对象csv_read，然后我们可以直接从csv_read对象中获取数据。

python中读取csv的方法有很多，下面讲一下常见的几种办法：

相关推荐：python基础教程

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

另一种方法用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爬虫视频教程